diff --git a/drivers/hab/hab.c b/drivers/hab/hab.c
index 512ff7e..0d71f5b 100644
--- a/drivers/hab/hab.c
+++ b/drivers/hab/hab.c
@@ -209,9 +209,9 @@
 	if (ops)
 		return ops;
 
-	if (cpu_is_mx25() || cpu_is_mx35())
+	if (IS_ENABLED(CONFIG_HABV3) && (cpu_is_mx25() || cpu_is_mx35()))
 		tmp = &imx_hab_ops_iim;
-	else if (cpu_is_mx6())
+	else if (IS_ENABLED(CONFIG_HABV4) && cpu_is_mx6())
 		tmp = &imx_hab_ops_ocotp;
 	else
 		return NULL;