diff --git a/Kbuild b/Kbuild
index 79c831f..825909f 100644
--- a/Kbuild
+++ b/Kbuild
@@ -116,7 +116,11 @@
 	CONFIG_WLAN_NAPI_DEBUG := n
 
 	# Flag to enable FW based TX Flow control
-	CONFIG_WLAN_TX_FLOW_CONTROL_V2 := n
+	ifeq ($(CONFIG_CNSS_EOS),y)
+		CONFIG_WLAN_TX_FLOW_CONTROL_V2 := y
+	else
+		CONFIG_WLAN_TX_FLOW_CONTROL_V2 := n
+	endif
 
 	# Flag to enable LRO (Large Receive Offload)
 	ifeq ($(CONFIG_CNSS_EOS), y)