diff --git a/Kbuild b/Kbuild
index 6b9f5a1..6d18d88 100644
--- a/Kbuild
+++ b/Kbuild
@@ -2363,6 +2363,9 @@
 
 ifeq ($(CONFIG_QCA6290_11AX), y)
 cppflags-y += -DQCA_WIFI_QCA6290_11AX
+endif
+
+ifeq ($(CONFIG_LITHIUM), y)
 cppflags-$(CONFIG_WLAN_TX_FLOW_CONTROL_V2) += -DQCA_AC_BASED_FLOW_CONTROL
 endif