}
 
 static int
-mt76u_refill_rx(struct mt76_dev *dev, struct urb *urb, int nsgs, gfp_t gfp)
+mt76u_refill_rx(struct mt76_dev *dev, struct mt76_queue *q,
+               struct urb *urb, int nsgs, gfp_t gfp)
 {
-       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
+       enum mt76_rxq_id qid = q - &dev->q_rx[MT_RXQ_MAIN];
 
-       if (dev->usb.sg_en)
+       if (qid == MT_RXQ_MAIN && dev->usb.sg_en)
                return mt76u_fill_rx_sg(dev, q, urb, nsgs, gfp);
 
        urb->transfer_buffer_length = q->buf_size;
 static int
 mt76u_rx_urb_alloc(struct mt76_dev *dev, struct mt76_queue_entry *e)
 {
+       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
        int err;
 
        err = mt76u_urb_alloc(dev, e, MT_RX_SG_MAX_SIZE);
        if (err)
                return err;
 
-       return mt76u_refill_rx(dev, e->urb, MT_RX_SG_MAX_SIZE,
+       return mt76u_refill_rx(dev, q, e->urb, MT_RX_SG_MAX_SIZE,
                               GFP_KERNEL);
 }
 
 
                count = mt76u_process_rx_entry(dev, urb, q->buf_size);
                if (count > 0) {
-                       err = mt76u_refill_rx(dev, urb, count, GFP_ATOMIC);
+                       err = mt76u_refill_rx(dev, q, urb, count, GFP_ATOMIC);
                        if (err < 0)
                                break;
                }