diff --git a/cnss2/main.c b/cnss2/main.c
index b18eb2d05a..9d2399c5a5 100644
--- a/cnss2/main.c
+++ b/cnss2/main.c
@@ -4474,7 +4474,7 @@ static void cnss_sram_dump_init(struct cnss_plat_data *plat_priv)
 }
 #endif
 
-#ifdef CONFIG_WCNSS_MEM_PRE_ALLOC
+#if IS_ENABLED(CONFIG_WCNSS_MEM_PRE_ALLOC)
 static void cnss_initialize_mem_pool(unsigned long device_id)
 {
 	cnss_initialize_prealloc_pool(device_id);
diff --git a/icnss2/main.c b/icnss2/main.c
index 44733b6813..a7841244b8 100644
--- a/icnss2/main.c
+++ b/icnss2/main.c
@@ -4620,7 +4620,7 @@ static void rproc_restart_level_notifier(void *data, struct rproc *rproc)
 	}
 }
 
-#ifdef CONFIG_WCNSS_MEM_PRE_ALLOC
+#if IS_ENABLED(CONFIG_WCNSS_MEM_PRE_ALLOC)
 static void icnss_initialize_mem_pool(unsigned long device_id)
 {
 	cnss_initialize_prealloc_pool(device_id);