int ret;
 
        cancel_delayed_work_sync(&dev->cal_work);
-       if (mt76_is_mmio(dev)) {
-               tasklet_disable(&dev->pre_tbtt_tasklet);
+       dev->beacon_ops->pre_tbtt_enable(dev, false);
+       if (mt76_is_mmio(dev))
                tasklet_disable(&dev->dfs_pd.dfs_tasklet);
-       }
 
        mt76_set_channel(&dev->mt76);
        ret = mt76x0_phy_set_channel(dev, chandef);
 
        if (mt76_is_mmio(dev)) {
                mt76x02_dfs_init_params(dev);
-               tasklet_enable(&dev->pre_tbtt_tasklet);
                tasklet_enable(&dev->dfs_pd.dfs_tasklet);
        }
+       dev->beacon_ops->pre_tbtt_enable(dev, true);
+
        mt76_txq_schedule_all(&dev->mt76);
 
        return ret;
 
        s8 tssi_dc;
 };
 
+struct mt76x02_beacon_ops {
+       void (*pre_tbtt_enable) (struct mt76x02_dev *, bool);
+       void (*beacon_enable) (struct mt76x02_dev *, bool);
+};
+
 struct mt76x02_dev {
        struct mt76_dev mt76; /* must be first */
 
        struct hrtimer pre_tbtt_timer;
        struct work_struct pre_tbtt_work;
 
+       const struct mt76x02_beacon_ops *beacon_ops;
+
        u32 aggr_stats[32];
 
        struct sk_buff *beacons[8];
 
              MT_BEACON_TIME_CFG_TIMER_EN;
        mt76_rmw(dev, MT_BEACON_TIME_CFG, reg, reg * en);
 
-       if (mt76_is_usb(dev))
-               return;
-
-       mt76_rmw_field(dev, MT_INT_TIMER_EN, MT_INT_TIMER_EN_PRE_TBTT_EN, en);
-       if (en)
-               mt76x02_irq_enable(dev, MT_INT_PRE_TBTT | MT_INT_TBTT);
-       else
-               mt76x02_irq_disable(dev, MT_INT_PRE_TBTT | MT_INT_TBTT);
+       dev->beacon_ops->beacon_enable(dev, en);
 }
 
 void mt76x02_mac_set_beacon_enable(struct mt76x02_dev *dev,
        u8 vif_idx = ((struct mt76x02_vif *)vif->drv_priv)->idx;
        struct sk_buff *skb = NULL;
 
-       if (mt76_is_mmio(dev))
-               tasklet_disable(&dev->pre_tbtt_tasklet);
-       else if (val)
+       dev->beacon_ops->pre_tbtt_enable(dev, false);
+
+       if (mt76_is_usb(dev))
                skb = ieee80211_beacon_get(mt76_hw(dev), vif);
 
        if (!dev->beacon_mask)
 
        __mt76x02_mac_set_beacon_enable(dev, vif_idx, val, skb);
 
-       if (mt76_is_mmio(dev))
-               tasklet_enable(&dev->pre_tbtt_tasklet);
+       dev->beacon_ops->pre_tbtt_enable(dev, true);
 }
 
 void mt76x02_init_beacon_config(struct mt76x02_dev *dev)
 
        spin_unlock_bh(&q->lock);
 }
 
+static void mt76x02e_pre_tbtt_enable(struct mt76x02_dev *dev, bool en)
+{
+       if (en)
+               tasklet_enable(&dev->pre_tbtt_tasklet);
+       else
+               tasklet_disable(&dev->pre_tbtt_tasklet);
+}
+
+static void mt76x02e_beacon_enable(struct mt76x02_dev *dev, bool en)
+{
+       mt76_rmw_field(dev, MT_INT_TIMER_EN, MT_INT_TIMER_EN_PRE_TBTT_EN, en);
+       if (en)
+               mt76x02_irq_enable(dev, MT_INT_PRE_TBTT | MT_INT_TBTT);
+       else
+               mt76x02_irq_disable(dev, MT_INT_PRE_TBTT | MT_INT_TBTT);
+}
+
 void mt76x02e_init_beacon_config(struct mt76x02_dev *dev)
 {
+       static const struct mt76x02_beacon_ops beacon_ops = {
+               .pre_tbtt_enable = mt76x02e_pre_tbtt_enable,
+               .beacon_enable = mt76x02e_beacon_enable,
+       };
+
+       dev->beacon_ops = &beacon_ops;
+
        /* Fire a pre-TBTT interrupt 8 ms before TBTT */
        mt76_rmw_field(dev, MT_INT_TIMER_CFG, MT_INT_TIMER_CFG_PRE_TBTT, 8 << 4);
        mt76_rmw_field(dev, MT_INT_TIMER_CFG, MT_INT_TIMER_CFG_GP_TIMER,
 
        return HRTIMER_NORESTART;
 }
 
+static void mt76x02u_pre_tbtt_enable(struct mt76x02_dev *dev, bool en)
+{
+}
+
+static void mt76x02u_beacon_enable(struct mt76x02_dev *dev, bool en)
+{
+}
+
 void mt76x02u_init_beacon_config(struct mt76x02_dev *dev)
 {
+       static const struct mt76x02_beacon_ops beacon_ops = {
+               .pre_tbtt_enable = mt76x02u_pre_tbtt_enable,
+               .beacon_enable = mt76x02u_beacon_enable,
+       };
+       dev->beacon_ops = &beacon_ops;
+
        hrtimer_init(&dev->pre_tbtt_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
        dev->pre_tbtt_timer.function = mt76x02u_pre_tbtt_interrupt;
        INIT_WORK(&dev->pre_tbtt_work, mt76x02u_pre_tbtt_work);
 
 
        mt76_set_channel(&dev->mt76);
 
+       dev->beacon_ops->pre_tbtt_enable(dev, false);
+
        mt76x2_mac_stop(dev, false);
 
        err = mt76x2u_phy_set_channel(dev, chandef);
        mt76x2_mac_resume(dev);
        mt76x02_edcca_init(dev, true);
 
+       dev->beacon_ops->pre_tbtt_enable(dev, true);
+
        clear_bit(MT76_RESET, &dev->mt76.state);
        mt76_txq_schedule_all(&dev->mt76);