diff --git a/cnss2/main.c b/cnss2/main.c
index d881bf20a4..12f326f90d 100644
--- a/cnss2/main.c
+++ b/cnss2/main.c
@@ -4440,7 +4440,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 eb546ce386..62983bbcd1 100644
--- a/icnss2/main.c
+++ b/icnss2/main.c
@@ -4621,7 +4621,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);