return NULL;
 }
 
+static struct ath12k *ath12k_mac_get_ar_by_chan(struct ieee80211_hw *hw,
+                                               struct ieee80211_channel *channel)
+{
+       struct ath12k_hw *ah = hw->priv;
+       struct ath12k *ar;
+       int i;
+
+       ar = ah->radio;
+
+       if (ah->num_radio == 1)
+               return ar;
+
+       for_each_ar(ah, ar, i) {
+               if (channel->center_freq >= ar->freq_low &&
+                   channel->center_freq <= ar->freq_high)
+                       return ar;
+       }
+       return NULL;
+}
+
+static struct ath12k *ath12k_get_ar_by_ctx(struct ieee80211_hw *hw,
+                                          struct ieee80211_chanctx_conf *ctx)
+{
+       if (!ctx)
+               return NULL;
+
+       return ath12k_mac_get_ar_by_chan(hw, ctx->def.chan);
+}
+
 static void ath12k_pdev_caps_update(struct ath12k *ar)
 {
        struct ath12k_base *ab = ar->ab;
 static int ath12k_mac_op_add_chanctx(struct ieee80211_hw *hw,
                                     struct ieee80211_chanctx_conf *ctx)
 {
-       struct ath12k_hw *ah = ath12k_hw_to_ah(hw);
        struct ath12k *ar;
        struct ath12k_base *ab;
 
-       ar = ath12k_ah_to_ar(ah, 0);
+       ar = ath12k_get_ar_by_ctx(hw, ctx);
+       if (!ar)
+               return -EINVAL;
+
        ab = ar->ab;
 
        ath12k_dbg(ab, ATH12K_DBG_MAC,
 static void ath12k_mac_op_remove_chanctx(struct ieee80211_hw *hw,
                                         struct ieee80211_chanctx_conf *ctx)
 {
-       struct ath12k_hw *ah = ath12k_hw_to_ah(hw);
        struct ath12k *ar;
        struct ath12k_base *ab;
 
-       ar = ath12k_ah_to_ar(ah, 0);
+       ar = ath12k_get_ar_by_ctx(hw, ctx);
+       if (!ar)
+               return;
+
        ab = ar->ab;
 
        ath12k_dbg(ab, ATH12K_DBG_MAC,
                                         struct ieee80211_chanctx_conf *ctx,
                                         u32 changed)
 {
-       struct ath12k_hw *ah = ath12k_hw_to_ah(hw);
        struct ath12k *ar;
        struct ath12k_base *ab;
 
-       ar = ath12k_ah_to_ar(ah, 0);
+       ar = ath12k_get_ar_by_ctx(hw, ctx);
+       if (!ar)
+               return;
+
        ab = ar->ab;
 
        mutex_lock(&ar->conf_mutex);
                    band->channels[i].center_freq > freq_high)
                        band->channels[i].flags |= IEEE80211_CHAN_DISABLED;
        }
+
+       ar->freq_low = freq_low;
+       ar->freq_high = freq_high;
 }
 
 static u32 ath12k_get_phy_id(struct ath12k *ar, u32 band)