diff --git a/package/mac80211/patches/570-ap_vlan_injection.patch b/package/mac80211/patches/570-ap_vlan_injection.patch
index e33cc6e2e5..581b4a0143 100644
--- a/package/mac80211/patches/570-ap_vlan_injection.patch
+++ b/package/mac80211/patches/570-ap_vlan_injection.patch
@@ -4,7 +4,7 @@
  		tx->sta = rcu_dereference(sdata->u.vlan.sta);
  		if (!tx->sta && sdata->dev->ieee80211_ptr->use_4addr)
  			return TX_DROP;
-+	} else if (sdata->vif.type == NL80211_IFTYPE_AP) {
++	} else if (info->flags & IEEE80211_TX_CTL_INJECTED) {
 +		tx->sta = sta_info_get_bss(sdata, hdr->addr1);
  	}
  	if (!tx->sta)