diff --git a/asoc/msm-pcm-routing-v2.c b/asoc/msm-pcm-routing-v2.c
index f20d9124fb6e..b1e4e355af32 100755
--- a/asoc/msm-pcm-routing-v2.c
+++ b/asoc/msm-pcm-routing-v2.c
@@ -42218,20 +42218,30 @@ static int get_drift_src_idx(int drift_src)
 {
 	if (drift_src == DRIFT_SRC_SW)
 		return DRIFT_SRC_SW;
-	else if ((drift_src >= AFE_PORT_ID_PRIMARY_TDM_RX)
+	else if (((drift_src >= AFE_PORT_ID_PRIMARY_TDM_RX)
 		&& (drift_src <= AFE_PORT_ID_PRIMARY_TDM_TX_7))
+		|| (drift_src == AFE_PORT_ID_PRIMARY_MI2S_RX)
+		|| (drift_src == AFE_PORT_ID_PRIMARY_MI2S_TX))
 		return DRIFT_SRC_AFE_PRI;
-	else if ((drift_src >= AFE_PORT_ID_SECONDARY_TDM_RX)
+	else if (((drift_src >= AFE_PORT_ID_SECONDARY_TDM_RX)
 		&& (drift_src <= AFE_PORT_ID_SECONDARY_TDM_TX_7))
+		|| (drift_src == AFE_PORT_ID_SECONDARY_MI2S_RX)
+		|| (drift_src == AFE_PORT_ID_SECONDARY_MI2S_TX))
 		return DRIFT_SRC_AFE_SEC;
-	else if ((drift_src >= AFE_PORT_ID_TERTIARY_TDM_RX)
+	else if (((drift_src >= AFE_PORT_ID_TERTIARY_TDM_RX)
 		&& (drift_src <= AFE_PORT_ID_TERTIARY_TDM_TX_7))
+		|| (drift_src == AFE_PORT_ID_TERTIARY_MI2S_RX)
+		|| (drift_src == AFE_PORT_ID_TERTIARY_MI2S_TX))
 		return DRIFT_SRC_AFE_TERT;
-	else if ((drift_src >= AFE_PORT_ID_QUATERNARY_TDM_RX)
+	else if (((drift_src >= AFE_PORT_ID_QUATERNARY_TDM_RX)
 		&& (drift_src <= AFE_PORT_ID_QUATERNARY_TDM_TX_7))
+		|| (drift_src == AFE_PORT_ID_QUATERNARY_MI2S_RX)
+		|| (drift_src == AFE_PORT_ID_QUATERNARY_MI2S_TX))
 		return DRIFT_SRC_AFE_QUAT;
-	else if ((drift_src >= AFE_PORT_ID_QUINARY_TDM_RX)
+	else if (((drift_src >= AFE_PORT_ID_QUINARY_TDM_RX)
 		&& (drift_src <= AFE_PORT_ID_QUINARY_TDM_TX_7))
+		|| (drift_src == AFE_PORT_ID_QUINARY_MI2S_RX)
+		|| (drift_src == AFE_PORT_ID_QUINARY_MI2S_TX))
 		return DRIFT_SRC_AFE_QUIN;
 	else
 		return -EINVAL;