static int mt76x02_poll_tx(struct napi_struct *napi, int budget)
 {
-       struct mt76x02_dev *dev = container_of(napi, struct mt76x02_dev, tx_napi);
+       struct mt76x02_dev *dev = container_of(napi, struct mt76x02_dev,
+                                              mt76.tx_napi);
        int i;
 
        mt76x02_mac_poll_tx_status(dev, false);
        if (ret)
                return ret;
 
-       netif_tx_napi_add(&dev->mt76.napi_dev, &dev->tx_napi, mt76x02_poll_tx,
-                         NAPI_POLL_WEIGHT);
-       napi_enable(&dev->tx_napi);
+       netif_tx_napi_add(&dev->mt76.napi_dev, &dev->mt76.tx_napi,
+                         mt76x02_poll_tx, NAPI_POLL_WEIGHT);
+       napi_enable(&dev->mt76.tx_napi);
 
        return 0;
 }
 
        if (intr & (MT_INT_TX_STAT | MT_INT_TX_DONE_ALL)) {
                mt76x02_irq_disable(dev, MT_INT_TX_DONE_ALL);
-               napi_schedule(&dev->tx_napi);
+               napi_schedule(&dev->mt76.tx_napi);
        }
 
        if (intr & MT_INT_GPTIMER) {
 void mt76x02_dma_cleanup(struct mt76x02_dev *dev)
 {
        tasklet_kill(&dev->mt76.tx_tasklet);
-       netif_napi_del(&dev->tx_napi);
+       netif_napi_del(&dev->mt76.tx_napi);
        mt76_dma_cleanup(&dev->mt76);
 }
 EXPORT_SYMBOL_GPL(mt76x02_dma_cleanup);
 
        tasklet_disable(&dev->mt76.pre_tbtt_tasklet);
        tasklet_disable(&dev->mt76.tx_tasklet);
-       napi_disable(&dev->tx_napi);
+       napi_disable(&dev->mt76.tx_napi);
 
        for (i = 0; i < ARRAY_SIZE(dev->mt76.napi); i++)
                napi_disable(&dev->mt76.napi[i]);
        clear_bit(MT76_RESET, &dev->mt76.state);
 
        tasklet_enable(&dev->mt76.tx_tasklet);
-       napi_enable(&dev->tx_napi);
-       napi_schedule(&dev->tx_napi);
+       napi_enable(&dev->mt76.tx_napi);
+       napi_schedule(&dev->mt76.tx_napi);
 
        tasklet_enable(&dev->mt76.pre_tbtt_tasklet);