diff --git a/qcom/khaje.dtsi b/qcom/khaje.dtsi
index e9e96265..eb633c12 100644
--- a/qcom/khaje.dtsi
+++ b/qcom/khaje.dtsi
@@ -2704,6 +2704,9 @@
 		phys = <&ufsphy_mem>;
 		phy-names = "ufsphy";
 
+		qcom,prime-mask = <0x80>;
+		qcom,silver-mask = <0x0f>;
+
 		limit-rx-hs-gear = <3>;
 		limit-tx-hs-gear = <3>;
 		lanes-per-direction = <2>;