}
 
 static int
-mt76u_process_rx_entry(struct mt76_dev *dev, struct urb *urb)
+mt76u_process_rx_entry(struct mt76_dev *dev, struct urb *urb,
+                      int buf_size)
 {
-       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
        u8 *data = urb->num_sgs ? sg_virt(&urb->sg[0]) : urb->transfer_buffer;
        int data_len = urb->num_sgs ? urb->sg[0].length : urb->actual_length;
        int len, nsgs = 1;
                return 0;
 
        data_len = min_t(int, len, data_len - MT_DMA_HDR_LEN);
-       skb = mt76u_build_rx_skb(data, data_len, q->buf_size);
+       skb = mt76u_build_rx_skb(data, data_len, buf_size);
        if (!skb)
                return 0;
 
                data_len = min_t(int, len, urb->sg[nsgs].length);
                skb_add_rx_frag(skb, skb_shinfo(skb)->nr_frags,
                                sg_page(&urb->sg[nsgs]),
-                               urb->sg[nsgs].offset,
-                               data_len, q->buf_size);
+                               urb->sg[nsgs].offset, data_len,
+                               buf_size);
                len -= data_len;
                nsgs++;
        }
        return usb_submit_urb(urb, GFP_ATOMIC);
 }
 
-static void mt76u_rx_tasklet(unsigned long data)
+static void
+mt76u_process_rx_queue(struct mt76_dev *dev, struct mt76_queue *q)
 {
-       struct mt76_dev *dev = (struct mt76_dev *)data;
+       int qid = q - &dev->q_rx[MT_RXQ_MAIN];
        struct urb *urb;
        int err, count;
 
-       rcu_read_lock();
-
        while (true) {
                urb = mt76u_get_next_rx_entry(dev);
                if (!urb)
                        break;
 
-               count = mt76u_process_rx_entry(dev, urb);
+               count = mt76u_process_rx_entry(dev, urb, q->buf_size);
                if (count > 0) {
                        err = mt76u_refill_rx(dev, urb, count, GFP_ATOMIC);
                        if (err < 0)
                }
                mt76u_submit_rx_buf(dev, urb);
        }
-       mt76_rx_poll_complete(dev, MT_RXQ_MAIN, NULL);
+       if (qid == MT_RXQ_MAIN)
+               mt76_rx_poll_complete(dev, MT_RXQ_MAIN, NULL);
+}
 
+static void mt76u_rx_tasklet(unsigned long data)
+{
+       struct mt76_dev *dev = (struct mt76_dev *)data;
+       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
+
+       rcu_read_lock();
+       mt76u_process_rx_queue(dev, q);
        rcu_read_unlock();
 }