diff --git a/net/mac80211/tx.c b/net/mac80211/tx.c
index b58c85abcb1be2f3a43028747cab1b71e99546fa..6cd3aeba870fd84532ca7ff62fc136c4f7fb2db2 100644
--- a/net/mac80211/tx.c
+++ b/net/mac80211/tx.c
@@ -3760,7 +3760,7 @@ struct sk_buff *ieee80211_tx_dequeue(struct ieee80211_hw *hw,
 encap_out:
 	IEEE80211_SKB_CB(skb)->control.vif = vif;
 
-	if (vif &&
+	if (tx.sta &&
 	    wiphy_ext_feature_isset(local->hw.wiphy, NL80211_EXT_FEATURE_AQL)) {
 		bool ampdu = txq->ac != IEEE80211_AC_VO;
 		u32 airtime;