spin_unlock_irqrestore(&dev->rx_lock, flags);
 }
 
-static void mt7601u_rx_tasklet(unsigned long data)
+static void mt7601u_rx_tasklet(struct tasklet_struct *t)
 {
-       struct mt7601u_dev *dev = (struct mt7601u_dev *) data;
+       struct mt7601u_dev *dev = from_tasklet(dev, t, rx_tasklet);
        struct mt7601u_dma_buf_rx *e;
 
        while ((e = mt7601u_rx_get_pending_entry(dev))) {
        spin_unlock_irqrestore(&dev->tx_lock, flags);
 }
 
-static void mt7601u_tx_tasklet(unsigned long data)
+static void mt7601u_tx_tasklet(struct tasklet_struct *t)
 {
-       struct mt7601u_dev *dev = (struct mt7601u_dev *) data;
+       struct mt7601u_dev *dev = from_tasklet(dev, t, tx_tasklet);
        struct sk_buff_head skbs;
        unsigned long flags;
 
 {
        int ret = -ENOMEM;
 
-       tasklet_init(&dev->tx_tasklet, mt7601u_tx_tasklet, (unsigned long) dev);
-       tasklet_init(&dev->rx_tasklet, mt7601u_rx_tasklet, (unsigned long) dev);
+       tasklet_setup(&dev->tx_tasklet, mt7601u_tx_tasklet);
+       tasklet_setup(&dev->rx_tasklet, mt7601u_rx_tasklet);
 
        ret = mt7601u_alloc_tx(dev);
        if (ret)