diff --git a/Kbuild b/Kbuild
index 66b9438..1d13fe3 100644
--- a/Kbuild
+++ b/Kbuild
@@ -1554,18 +1554,9 @@
 cppflags-$(CONFIG_FEATURE_MEMDUMP_ENABLE) += -DWLAN_FEATURE_MEMDUMP_ENABLE
 cppflags-$(CONFIG_FEATURE_FW_LOG_PARSING) += -DFEATURE_FW_LOG_PARSING
 
-ifeq ($(CONFIG_CNSS), y)
-ifeq ($(CONFIG_CNSS_SDIO), y)
-cppflags-y += -DCONFIG_PLD_SDIO_CNSS
-else
-cppflags-y += -DCONFIG_PLD_PCIE_CNSS
-endif
-endif
-
-ifeq ($(CONFIG_CNSS2), y)
-cppflags-y += -DCONFIG_PLD_PCIE_CNSS
-cppflags-y += -DCONFIG_PLD_PCIE_INIT
-endif
+cppflags-$(CONFIG_PLD_SDIO_CNSS_FLAG) += -DCONFIG_PLD_SDIO_CNSS
+cppflags-$(CONFIG_PLD_PCIE_CNSS_FLAG) += -DCONFIG_PLD_PCIE_CNSS
+cppflags-$(CONFIG_PLD_PCIE_INIT_FLAG) += -DCONFIG_PLD_PCIE_INIT
 
 #Enable NL80211 test mode
 cppflags-$(CONFIG_NL80211_TESTMODE) += -DWLAN_NL80211_TESTMODE