bool is_bmc;
        u16 seq;
 
+       if (tx_req->sta)
+               desc_info->mlo = tx_req->sta->mlo;
+       else if (tx_req->vif)
+               desc_info->mlo = ieee80211_vif_is_mld(tx_req->vif);
+
        seq = (le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_SEQ) >> 4;
        if (tx_req->tx_type != RTW89_CORE_TX_TYPE_FWCMD) {
                tx_type = rtw89_core_get_tx_type(rtwdev, skb);
 
                addr_cam = rtw89_get_addr_cam_of(tx_req->rtwvif_link,
                                                 tx_req->rtwsta_link);
-               if (addr_cam->valid)
+               if (addr_cam->valid && desc_info->mlo)
                        upd_wlan_hdr = true;
        }
        is_bmc = (is_broadcast_ether_addr(hdr->addr1) ||
        }
 
        tx_req.skb = skb;
+       tx_req.vif = vif;
+       tx_req.sta = sta;
        tx_req.rtwvif_link = rtwvif_link;
        tx_req.rtwsta_link = rtwsta_link;
 
 
        bool stbc;
        bool ldpc;
        bool upd_wlan_hdr;
+       bool mlo;
 };
 
 struct rtw89_core_tx_request {
        enum rtw89_core_tx_type tx_type;
 
        struct sk_buff *skb;
+       struct ieee80211_vif *vif;
+       struct ieee80211_sta *sta;
        struct rtw89_vif_link *rtwvif_link;
        struct rtw89_sta_link *rtwsta_link;
        struct rtw89_tx_desc_info desc_info;