diff --git a/cnss2/main.c b/cnss2/main.c
index 12f326f90d..d881bf20a4 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
 
-#if IS_ENABLED(CONFIG_WCNSS_MEM_PRE_ALLOC)
+#ifdef 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 62983bbcd1..eb546ce386 100644
--- a/icnss2/main.c
+++ b/icnss2/main.c
@@ -4621,7 +4621,7 @@ static void rproc_restart_level_notifier(void *data, struct rproc *rproc)
 	}
 }
 
-#if IS_ENABLED(CONFIG_WCNSS_MEM_PRE_ALLOC)
+#ifdef CONFIG_WCNSS_MEM_PRE_ALLOC
 static void icnss_initialize_mem_pool(unsigned long device_id)
 {
 	cnss_initialize_prealloc_pool(device_id);