static inline int
 mt76u_bulk_msg(struct mt76_dev *dev, void *data, int len, int *actual_len,
-              int timeout)
+              int timeout, int ep)
 {
        struct usb_interface *uintf = to_usb_interface(dev->dev);
        struct usb_device *udev = interface_to_usbdev(uintf);
        unsigned int pipe;
 
        if (actual_len)
-               pipe = usb_rcvbulkpipe(udev, usb->in_ep[MT_EP_IN_CMD_RESP]);
+               pipe = usb_rcvbulkpipe(udev, usb->in_ep[ep]);
        else
-               pipe = usb_sndbulkpipe(udev, usb->out_ep[MT_EP_OUT_INBAND_CMD]);
+               pipe = usb_sndbulkpipe(udev, usb->out_ep[ep]);
 
        return usb_bulk_msg(udev, pipe, data, len, actual_len, timeout);
 }
 
        u32 rxfce;
 
        for (i = 0; i < 5; i++) {
-               ret = mt76u_bulk_msg(dev, data, MCU_RESP_URB_SIZE, &len, 300);
+               ret = mt76u_bulk_msg(dev, data, MCU_RESP_URB_SIZE, &len,
+                                    300, MT_EP_IN_CMD_RESP);
                if (ret == -ETIMEDOUT)
                        continue;
                if (ret)
        if (ret)
                return ret;
 
-       ret = mt76u_bulk_msg(dev, skb->data, skb->len, NULL, 500);
+       ret = mt76u_bulk_msg(dev, skb->data, skb->len, NULL, 500,
+                            MT_EP_OUT_INBAND_CMD);
        if (ret)
                return ret;
 
 
        data_len = MT_CMD_HDR_LEN + len + sizeof(info);
 
-       err = mt76u_bulk_msg(&dev->mt76, data, data_len, NULL, 1000);
+       err = mt76u_bulk_msg(&dev->mt76, data, data_len, NULL, 1000,
+                            MT_EP_OUT_INBAND_CMD);
        if (err) {
                dev_err(dev->mt76.dev, "firmware upload failed: %d\n", err);
                return err;