*/
 
        if (changed & BSS_CHANGED_BEACON_ENABLED) {
-               if (info->enable_beacon) {
-                       mt7615_mcu_set_bss_info(dev, vif, 1);
-                       mt7615_mcu_add_wtbl_bmc(dev, vif);
-                       mt7615_mcu_set_sta_rec_bmc(dev, vif, 1);
-                       mt7615_mcu_set_bcn(dev, vif, 1);
-               } else {
-                       mt7615_mcu_set_sta_rec_bmc(dev, vif, 0);
-                       mt7615_mcu_del_wtbl_bmc(dev, vif);
-                       mt7615_mcu_set_bss_info(dev, vif, 0);
-                       mt7615_mcu_set_bcn(dev, vif, 0);
-               }
+               mt7615_mcu_set_bss_info(dev, vif, info->enable_beacon);
+               mt7615_mcu_wtbl_bmc(dev, vif, info->enable_beacon);
+               mt7615_mcu_set_sta_rec_bmc(dev, vif, info->enable_beacon);
+               mt7615_mcu_set_bcn(dev, vif, info->enable_beacon);
        }
 
        mutex_unlock(&dev->mt76.mutex);
 
                                     &wtbl_sec_key, buf_len);
 }
 
-int mt7615_mcu_add_wtbl_bmc(struct mt7615_dev *dev,
-                           struct ieee80211_vif *vif)
+static int
+mt7615_mcu_add_wtbl_bmc(struct mt7615_dev *dev,
+                       struct mt7615_vif *mvif)
 {
-       struct mt7615_vif *mvif = (struct mt7615_vif *)vif->drv_priv;
        struct {
                struct wtbl_generic g_wtbl;
                struct wtbl_rx rx_wtbl;
                                     sizeof(struct wtbl_rx));
 }
 
-int mt7615_mcu_del_wtbl_bmc(struct mt7615_dev *dev, struct ieee80211_vif *vif)
+int mt7615_mcu_wtbl_bmc(struct mt7615_dev *dev,
+                       struct ieee80211_vif *vif, bool enable)
 {
        struct mt7615_vif *mvif = (struct mt7615_vif *)vif->drv_priv;
 
+       if (enable)
+               return mt7615_mcu_add_wtbl_bmc(dev, mvif);
+
        return __mt7615_mcu_set_wtbl(dev, mvif->sta.wcid.idx,
                                     WTBL_RESET_AND_SET, 0, NULL, 0);
 }
 
 void mt7615_mcu_set_rates(struct mt7615_dev *dev, struct mt7615_sta *sta,
                          struct ieee80211_tx_rate *probe_rate,
                          struct ieee80211_tx_rate *rates);
-int mt7615_mcu_add_wtbl_bmc(struct mt7615_dev *dev, struct ieee80211_vif *vif);
-int mt7615_mcu_del_wtbl_bmc(struct mt7615_dev *dev, struct ieee80211_vif *vif);
+int mt7615_mcu_wtbl_bmc(struct mt7615_dev *dev, struct ieee80211_vif *vif,
+                       bool enable);
 int mt7615_mcu_add_wtbl(struct mt7615_dev *dev, struct ieee80211_vif *vif,
                        struct ieee80211_sta *sta);
 int mt7615_mcu_del_wtbl(struct mt7615_dev *dev, struct ieee80211_vif *vif,