diff --git a/Kbuild b/Kbuild
index df242803fc..0fa71dc2f1 100644
--- a/Kbuild
+++ b/Kbuild
@@ -2706,6 +2706,7 @@ endif
 
 ifeq ($(CONFIG_LITHIUM), y)
 cppflags-$(CONFIG_WLAN_TX_FLOW_CONTROL_V2) += -DQCA_AC_BASED_FLOW_CONTROL
+cppflags-y += -DHAL_DISABLE_NON_BA_2K_JUMP_ERROR
 endif
 
 cppflags-$(CONFIG_WLAN_CLD_PM_QOS) += -DCLD_PM_QOS