diff --git a/drivers/soc/qcom/hab/hab.c b/drivers/soc/qcom/hab/hab.c
index b4ce657edae0..65935f6eb48f 100644
--- a/drivers/soc/qcom/hab/hab.c
+++ b/drivers/soc/qcom/hab/hab.c
@@ -1315,6 +1315,8 @@ static int __init hab_init(void)
 	int result;
 	dev_t dev;
 
+	place_marker("M - HAB INIT Start");
+
 	result = alloc_chrdev_region(&hab_driver.major, 0, 1, "hab");
 
 	if (result < 0) {
@@ -1372,6 +1374,8 @@ static int __init hab_init(void)
 
 	hab_stat_init(&hab_driver);
 
+	place_marker("M - HAB INIT End");
+
 	return result;
 
 err:
diff --git a/drivers/soc/qcom/hab/hab.h b/drivers/soc/qcom/hab/hab.h
index f381303f3ca8..c4e8eaa368e3 100644
--- a/drivers/soc/qcom/hab/hab.h
+++ b/drivers/soc/qcom/hab/hab.h
@@ -45,6 +45,7 @@
 #include <linux/reboot.h>
 #include <linux/kobject.h>
 #include <linux/sysfs.h>
+#include <soc/qcom/boot_stats.h>
 
 enum hab_payload_type {
 	HAB_PAYLOAD_TYPE_MSG = 0x0,