static void mt76u_complete_rx(struct urb *urb)
 {
-       struct mt76_dev *dev = urb->context;
-       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
+       struct mt76_dev *dev = dev_get_drvdata(&urb->dev->dev);
+       struct mt76_queue *q = urb->context;
        unsigned long flags;
 
        trace_rx_urb(dev, urb);
 }
 
 static int
-mt76u_submit_rx_buf(struct mt76_dev *dev, struct urb *urb)
+mt76u_submit_rx_buf(struct mt76_dev *dev, enum mt76_rxq_id qid,
+                   struct urb *urb)
 {
-       mt76u_fill_bulk_urb(dev, USB_DIR_IN, MT_EP_IN_PKT_RX, urb,
-                           mt76u_complete_rx, dev);
+       int ep = qid == MT_RXQ_MAIN ? MT_EP_IN_PKT_RX : MT_EP_IN_CMD_RESP;
+
+       mt76u_fill_bulk_urb(dev, USB_DIR_IN, ep, urb,
+                           mt76u_complete_rx, &dev->q_rx[qid]);
        trace_submit_urb(dev, urb);
 
        return usb_submit_urb(urb, GFP_ATOMIC);
                        if (err < 0)
                                break;
                }
-               mt76u_submit_rx_buf(dev, urb);
+               mt76u_submit_rx_buf(dev, qid, urb);
        }
        if (qid == MT_RXQ_MAIN)
                mt76_rx_poll_complete(dev, MT_RXQ_MAIN, NULL);
 
        spin_lock_irqsave(&q->lock, flags);
        for (i = 0; i < q->ndesc; i++) {
-               err = mt76u_submit_rx_buf(dev, q->entry[i].urb);
+               err = mt76u_submit_rx_buf(dev, MT_RXQ_MAIN, q->entry[i].urb);
                if (err < 0)
                        break;
        }