static int ath12k_start_vdev_delay(struct ath12k *ar,
                                   struct ath12k_vif *arvif);
+static void ath12k_mac_stop(struct ath12k *ar);
 
 static const char *ath12k_mac_phymode_str(enum wmi_phy_mode mode)
 {
        return ret;
 }
 
+static void ath12k_drain_tx(struct ath12k_hw *ah)
+{
+       struct ath12k *ar;
+       int i;
+
+       for_each_ar(ah, ar, i)
+               ath12k_mac_drain_tx(ar);
+}
+
 static int ath12k_mac_op_start(struct ieee80211_hw *hw)
 {
        struct ath12k_hw *ah = ath12k_hw_to_ah(hw);
-       struct ath12k *ar = ath12k_ah_to_ar(ah, 0);
-       struct ath12k_base *ab = ar->ab;
-       int ret;
+       struct ath12k *ar;
+       int ret, i;
 
-       ath12k_mac_drain_tx(ar);
+       ath12k_drain_tx(ah);
 
-       ret = ath12k_mac_start(ar);
-       if (ret) {
-               ath12k_err(ab, "fail to start mac operations in pdev idx %d ret %d\n",
-                          ar->pdev_idx, ret);
-               return ret;
+       for_each_ar(ah, ar, i) {
+               ret = ath12k_mac_start(ar);
+               if (ret) {
+                       ath12k_err(ar->ab, "fail to start mac operations in pdev idx %d ret %d\n",
+                                  ar->pdev_idx, ret);
+                       goto fail_start;
+               }
        }
 
        return 0;
+fail_start:
+       for (; i > 0; i--) {
+               ar = ath12k_ah_to_ar(ah, i - 1);
+               ath12k_mac_stop(ar);
+       }
+       return ret;
 }
 
 int ath12k_mac_rfkill_config(struct ath12k *ar)
 static void ath12k_mac_op_stop(struct ieee80211_hw *hw)
 {
        struct ath12k_hw *ah = ath12k_hw_to_ah(hw);
-       struct ath12k *ar = ath12k_ah_to_ar(ah, 0);
+       struct ath12k *ar;
+       int i;
 
-       ath12k_mac_drain_tx(ar);
+       ath12k_drain_tx(ah);
 
-       ath12k_mac_stop(ar);
+       for_each_ar(ah, ar, i)
+               ath12k_mac_stop(ar);
 }
 
 static u8