diff --git a/dsp/q6afe.c b/dsp/q6afe.c
index 5f54be086daf..5320ef4937c7 100644
--- a/dsp/q6afe.c
+++ b/dsp/q6afe.c
@@ -4098,6 +4098,20 @@ int afe_get_port_index(u16 port_id)
 		return IDX_AFE_PORT_ID_INT6_MI2S_RX;
 	case AFE_PORT_ID_INT6_MI2S_TX:
 		return IDX_AFE_PORT_ID_INT6_MI2S_TX;
+	case AFE_PORT_ID_VA_CODEC_DMA_TX_0:
+		return IDX_AFE_PORT_ID_VA_CODEC_DMA_TX_0;
+	case AFE_PORT_ID_VA_CODEC_DMA_TX_1:
+		return IDX_AFE_PORT_ID_VA_CODEC_DMA_TX_1;
+	case AFE_PORT_ID_WSA_CODEC_DMA_RX_0:
+		return IDX_AFE_PORT_ID_WSA_CODEC_DMA_RX_0;
+	case AFE_PORT_ID_WSA_CODEC_DMA_TX_0:
+		return IDX_AFE_PORT_ID_WSA_CODEC_DMA_TX_0;
+	case AFE_PORT_ID_WSA_CODEC_DMA_RX_1:
+		return IDX_AFE_PORT_ID_WSA_CODEC_DMA_RX_1;
+	case AFE_PORT_ID_WSA_CODEC_DMA_TX_1:
+		return IDX_AFE_PORT_ID_WSA_CODEC_DMA_TX_1;
+	case AFE_PORT_ID_WSA_CODEC_DMA_TX_2:
+		return IDX_AFE_PORT_ID_WSA_CODEC_DMA_TX_2;
 	default:
 		pr_err("%s: port 0x%x\n", __func__, port_id);
 		return -EINVAL;