diff --git a/net/mac80211/agg-tx.c b/net/mac80211/agg-tx.c
index 92e5812daf892d1434304c5ab3948bed6b1282bc..433009b3e8ab7a1e7b62e0df1c3397315b0cac39 100644
--- a/net/mac80211/agg-tx.c
+++ b/net/mac80211/agg-tx.c
@@ -599,7 +599,8 @@ int ieee80211_start_tx_ba_session(struct ieee80211_sta *pubsta, u16 tid,
 		return -EINVAL;
 
 	if (!pubsta->ht_cap.ht_supported &&
-	    sta->sdata->vif.bss_conf.chandef.chan->band != NL80211_BAND_6GHZ)
+	    !pubsta->vht_cap.vht_supported &&
+	    !pubsta->he_cap.has_he)
 		return -EINVAL;
 
 	if (WARN_ON_ONCE(!local->ops->ampdu_action))