diff --git a/qcom/shima-pmic-overlay.dtsi b/qcom/shima-pmic-overlay.dtsi
index f09283d5d130..7a3fa06e69f9 100644
--- a/qcom/shima-pmic-overlay.dtsi
+++ b/qcom/shima-pmic-overlay.dtsi
@@ -501,3 +501,11 @@
 	io-channels = <&pmk8350_vadc PMR735B_ADC7_DIE_TEMP>;
 	io-channel-names = "thermal";
 };
+
+&soc {
+	reboot_reason {
+		compatible = "qcom,reboot-reason";
+		nvmem-cells = <&restart_reason>;
+		nvmem-cell-names = "restart_reason";
+	};
+};
diff --git a/qcom/shima.dtsi b/qcom/shima.dtsi
index ed3798d034b0..74f816999bf1 100644
--- a/qcom/shima.dtsi
+++ b/qcom/shima.dtsi
@@ -417,6 +417,7 @@
 	firmware: firmware {
 		scm {
 			compatible = "qcom,scm";
+			qcom,dload-mode = <&tcsr 0x13000>;
 		};
 
 		android {
@@ -987,6 +988,11 @@
 		#hwlock-cells = <1>;
 	};
 
+	tcsr: syscon@1fc0000 {
+		compatible = "syscon";
+		reg = <0x1fc0000 0x30000>;
+	};
+
 	smem: qcom,smem {
 		compatible = "qcom,smem";
 		memory-region = <&smem_mem>;
@@ -1787,10 +1793,8 @@
 				0x18040060 0x18050060 0x18060060 0x18070060>;
 	};
 
-	restart: restart@c264000 {
-		compatible = "qcom,pshold";
-		reg = <0xc264000 0x4>, <0x1fd3000 0x4>;
-		reg-names = "pshold-base", "tcsr-boot-misc-detect";
+	dload_mode {
+		compatible = "qcom,dload-mode";
 	};
 
 	cache-controller@9200000 {