int ret;
 
        cancel_delayed_work_sync(&dev->cal_work);
+       tasklet_disable(&dev->pre_tbtt_tasklet);
 
        mt76_set_channel(&dev->mt76);
        ret = mt76x0_phy_set_channel(dev, chandef);
        mt76_rr(dev, MT_CH_IDLE);
        mt76_rr(dev, MT_CH_BUSY);
 
+       tasklet_enable(&dev->pre_tbtt_tasklet);
        mt76_txq_schedule_all(&dev->mt76);
 
        return ret;
 
 static void mt76x0e_cleanup(struct mt76x02_dev *dev)
 {
        clear_bit(MT76_STATE_INITIALIZED, &dev->mt76.state);
+       tasklet_disable(&dev->pre_tbtt_tasklet);
        mt76x0_chip_onoff(dev, false, false);
        mt76x0e_stop_hw(dev);
        mt76x02_dma_cleanup(dev);
 
                           struct mt76_wcid *wcid, struct ieee80211_sta *sta,
                           u32 *tx_info);
 
-void mt76x02_pre_tbtt_tasklet(unsigned long arg);
-
 extern const u16 mt76x02_beacon_offsets[16];
 void mt76x02_set_beacon_offsets(struct mt76x02_dev *dev);
 void mt76x02_set_irq_mask(struct mt76x02_dev *dev, u32 clear, u32 set);
 
                       MT_BEACON_TIME_CFG_INTVAL, timer_val);
 }
 
-void mt76x02_pre_tbtt_tasklet(unsigned long arg)
+static void mt76x02_pre_tbtt_tasklet(unsigned long arg)
 {
        struct mt76x02_dev *dev = (struct mt76x02_dev *)arg;
        struct mt76_queue *q = &dev->mt76.q_tx[MT_TXQ_PSD];
        }
        spin_unlock_bh(&q->lock);
 }
-EXPORT_SYMBOL_GPL(mt76x02_pre_tbtt_tasklet);
 
 static int
 mt76x02_init_tx_queue(struct mt76x02_dev *dev, struct mt76_queue *q,
                return -ENOMEM;
 
        tasklet_init(&dev->tx_tasklet, mt76x02_tx_tasklet, (unsigned long) dev);
+       tasklet_init(&dev->pre_tbtt_tasklet, mt76x02_pre_tbtt_tasklet,
+                    (unsigned long)dev);
+
        kfifo_init(&dev->txstatus_fifo, status_fifo, fifo_size);
 
        mt76_dma_attach(&dev->mt76);
 
 {
        int ret;
 
-       tasklet_init(&dev->pre_tbtt_tasklet, mt76x02_pre_tbtt_tasklet,
-                    (unsigned long) dev);
-
        mt76x02_dma_disable(dev);
        mt76x2_reset_wlan(dev, true);
        mt76x2_power_on(dev);