diff --git a/Kbuild b/Kbuild
index cf403ba..7414ebd 100644
--- a/Kbuild
+++ b/Kbuild
@@ -26,6 +26,9 @@
 #Flag to enable SAE
     CONFIG_WLAN_FEATURE_SAE := y
 
+#Flag to enable SW PTA
+    CONFIG_WLAN_FEATURE_SW_PTA := y
+
 #Flag to enable OWE
     CONFIG_WLAN_AKM_SUITE_OWE := y
 
@@ -691,6 +694,10 @@
 CDEFINES += -DWLAN_FEATURE_SAE
 endif
 
+ifeq ($(CONFIG_WLAN_FEATURE_SW_PTA),y)
+CDEFINES += -DFEATURE_WLAN_SW_PTA
+endif
+
 ifeq ($(CONFIG_WLAN_AKM_SUITE_OWE),y)
 CDEFINES += -DWLAN_AKM_SUITE_OWE
 endif