diff --git a/qcom/waipio-pmic-overlay.dtsi b/qcom/waipio-pmic-overlay.dtsi
index 357fd8b5..3ca5c878 100644
--- a/qcom/waipio-pmic-overlay.dtsi
+++ b/qcom/waipio-pmic-overlay.dtsi
@@ -63,6 +63,15 @@
 	};
 };
 
+&pm8350b_gpios {
+	pm8350b_rear_tof_therm {
+		pm8350b_rear_tof_therm_default: pm8350b_rear_tof_therm_default {
+			pins = "gpio2";
+			bias-high-impedance;
+		};
+	};
+};
+
 &pm8350c_gpios {
 	lcd_backlight_ctrl {
 		lcd_backlight_ctrl_default: lcd_backlight_ctrl_default {
@@ -108,7 +117,7 @@
 
 &pmk8350_vadc {
 	pinctrl-names = "default";
-	pinctrl-0 = <&pm8350_rear_tof_therm_default>;
+	pinctrl-0 = <&pm8350b_rear_tof_therm_default>;
 
 	pm8350b_chg_temp {
 		reg = <PM8350B_ADC7_CHG_TEMP>;