diff --git a/Kbuild b/Kbuild
index bbce42fd0f86..ebdbbf19413a 100644
--- a/Kbuild
+++ b/Kbuild
@@ -107,7 +107,7 @@ ifeq ($(KERNEL_BUILD), 0)
 	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