#include <dt-bindings/pinctrl/qcom,pmic-gpio.h>
 #include <dt-bindings/sound/qcom,q6afe.h>
 #include <dt-bindings/sound/qcom,q6asm.h>
+#include <dt-bindings/sound/qcom,wcd9335.h>
 
 /*
  * GPIO name legend: proper name = the GPIO line is used as GPIO
        };
 
                codec {
-                       sound-dai = <&wcd9335 6>;
+                       sound-dai = <&wcd9335 AIF4_PB>;
                };
        };
 
                };
 
                codec {
-                       sound-dai = <&wcd9335 1>;
+                       sound-dai = <&wcd9335 AIF1_CAP>;
                };
        };
 };
 
 #include "msm8996-xiaomi-common.dtsi"
 #include <dt-bindings/sound/qcom,q6afe.h>
 #include <dt-bindings/sound/qcom,q6asm.h>
+#include <dt-bindings/sound/qcom,wcd9335.h>
 #include <dt-bindings/input/ti-drv260x.h>
 
 / {
                };
 
                codec {
-                       sound-dai = <&wcd9335 6>;
+                       sound-dai = <&wcd9335 AIF4_PB>;
                };
        };
 
                };
 
                codec {
-                       sound-dai = <&wcd9335 1>;
+                       sound-dai = <&wcd9335 AIF1_CAP>;
                };
        };
 };
 
 #include "pmi8996.dtsi"
 #include <dt-bindings/sound/qcom,q6afe.h>
 #include <dt-bindings/sound/qcom,q6asm.h>
+#include <dt-bindings/sound/qcom,wcd9335.h>
 
 / {
        model = "Xiaomi Mi Note 2";
                };
 
                codec {
-                       sound-dai = <&wcd9335 6>;
+                       sound-dai = <&wcd9335 AIF4_PB>;
                };
        };
 
                };
 
                codec {
-                       sound-dai = <&wcd9335 1>;
+                       sound-dai = <&wcd9335 AIF1_CAP>;
                };
        };
 };