static void
 ieee80211_update_from_he_6ghz_capa(const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
-                                  struct sta_info *sta)
+                                  struct sta_info *sta, unsigned int link_id)
 {
        enum ieee80211_smps_mode smps_mode;
 
                break;
        }
 
-       sta->sta.deflink.he_6ghz_capa = *he_6ghz_capa;
+       sta->sta.link[link_id]->he_6ghz_capa = *he_6ghz_capa;
 }
 
 static void ieee80211_he_mcs_disable(__le16 *he_mcs)
                                  struct ieee80211_supported_band *sband,
                                  const u8 *he_cap_ie, u8 he_cap_len,
                                  const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
-                                 struct sta_info *sta)
+                                 struct sta_info *sta, unsigned int link_id)
 {
-       struct ieee80211_sta_he_cap *he_cap = &sta->sta.deflink.he_cap;
+       struct ieee80211_sta_he_cap *he_cap = &sta->sta.link[link_id]->he_cap;
        struct ieee80211_sta_he_cap own_he_cap;
        struct ieee80211_he_cap_elem *he_cap_ie_elem = (void *)he_cap_ie;
        u8 he_ppe_size;
 
        he_cap->has_he = true;
 
-       sta->deflink.cur_max_bandwidth = ieee80211_sta_cap_rx_bw(sta, 0);
-       sta->sta.deflink.bandwidth = ieee80211_sta_cur_vht_bw(sta, 0);
+       sta->link[link_id]->cur_max_bandwidth =
+               ieee80211_sta_cap_rx_bw(sta, link_id);
+       sta->sta.link[link_id]->bandwidth =
+               ieee80211_sta_cur_vht_bw(sta, link_id);
 
        if (sband->band == NL80211_BAND_6GHZ && he_6ghz_capa)
-               ieee80211_update_from_he_6ghz_capa(he_6ghz_capa, sta);
+               ieee80211_update_from_he_6ghz_capa(he_6ghz_capa, sta, link_id);
 
        ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80,
                                      &he_cap->he_mcs_nss_supp.rx_mcs_80,
 
                                  struct ieee80211_supported_band *sband,
                                  const u8 *he_cap_ie, u8 he_cap_len,
                                  const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
-                                 struct sta_info *sta);
+                                 struct sta_info *sta, unsigned int link_id);
 void
 ieee80211_he_spr_ie_to_bss_conf(struct ieee80211_vif *vif,
                                const struct ieee80211_he_spr *he_spr_ie_elem);