struct ieee80211_vif *vif,
                             u8 rcpi, u8 sta_state)
 {
-       struct cfg80211_chan_def *chandef = &mphy->chandef;
+       struct mt76_vif *mvif = (struct mt76_vif *)vif->drv_priv;
+       struct cfg80211_chan_def *chandef = mvif->ctx ?
+                                           &mvif->ctx->def : &mphy->chandef;
        enum nl80211_band band = chandef->chan->band;
        struct mt76_dev *dev = mphy->dev;
        struct sta_rec_ra_info *ra_info;
 const struct ieee80211_sta_he_cap *
 mt76_connac_get_he_phy_cap(struct mt76_phy *phy, struct ieee80211_vif *vif)
 {
-       enum nl80211_band band = phy->chandef.chan->band;
+       struct mt76_vif *mvif = (struct mt76_vif *)vif->drv_priv;
+       struct cfg80211_chan_def *chandef = mvif->ctx ?
+                                           &mvif->ctx->def : &phy->chandef;
+       enum nl80211_band band = chandef->chan->band;
        struct ieee80211_supported_band *sband;
 
        sband = phy->hw->wiphy->bands[band];