diff --git a/Kbuild b/Kbuild
index bbce42f..ebdbbf1 100644
--- a/Kbuild
+++ b/Kbuild
@@ -107,7 +107,7 @@
 	ifeq ($(CONFIG_MOBILE_ROUTER), y)
 	CONFIG_QCACLD_FEATURE_GREEN_AP := y
 	endif
-	ifeq ($(CONFIG_ARCH_MSMCOBALT), y)
+	ifeq ($(CONFIG_ARCH_MSM8998), y)
 	CONFIG_QCACLD_FEATURE_GREEN_AP := y
 	endif