return 0;
 }
 
-int mt76_dma_tx_queue_skb(struct mt76_dev *dev, struct mt76_queue *q,
+int mt76_dma_tx_queue_skb(struct mt76_dev *dev, enum mt76_txq_id qid,
                          struct sk_buff *skb, struct mt76_wcid *wcid,
                          struct ieee80211_sta *sta)
 {
+       struct mt76_queue *q = &dev->q_tx[qid];
        struct mt76_queue_entry e;
        struct mt76_txwi_cache *t;
        struct mt76_queue_buf buf[32];
 
                       struct mt76_queue_buf *buf, int nbufs, u32 info,
                       struct sk_buff *skb, void *txwi);
 
-       int (*tx_queue_skb)(struct mt76_dev *dev, struct mt76_queue *q,
+       int (*tx_queue_skb)(struct mt76_dev *dev, enum mt76_txq_id qid,
                            struct sk_buff *skb, struct mt76_wcid *wcid,
                            struct ieee80211_sta *sta);
 
        return ((void *) IEEE80211_SKB_CB(skb)->status.status_driver_data);
 }
 
-int mt76_dma_tx_queue_skb(struct mt76_dev *dev, struct mt76_queue *q,
+int mt76_dma_tx_queue_skb(struct mt76_dev *dev, enum mt76_txq_id qid,
                          struct sk_buff *skb, struct mt76_wcid *wcid,
                          struct ieee80211_sta *sta);
 
 
        if (!skb)
                return;
 
-       mt76_dma_tx_queue_skb(&dev->mt76, &dev->mt76.q_tx[MT_TXQ_BEACON], skb,
+       mt76_dma_tx_queue_skb(&dev->mt76, MT_TXQ_BEACON, skb,
                              &mvif->sta.wcid, NULL);
 
        spin_lock_bh(&dev->ps_lock);
                struct ieee80211_vif *vif = info->control.vif;
                struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
 
-               mt76_dma_tx_queue_skb(&dev->mt76, q, skb, &mvif->sta.wcid,
-                                     NULL);
+               mt76_dma_tx_queue_skb(&dev->mt76, MT_TXQ_CAB, skb,
+                                     &mvif->sta.wcid, NULL);
        }
        mt76_queue_kick(dev, q);
        spin_unlock_bh(&q->lock);
 
                struct ieee80211_vif *vif = info->control.vif;
                struct mt76x02_vif *mvif = (struct mt76x02_vif *)vif->drv_priv;
 
-               mt76_dma_tx_queue_skb(&dev->mt76, q, skb, &mvif->group_wcid,
-                                     NULL);
+               mt76_dma_tx_queue_skb(&dev->mt76, MT_TXQ_PSD, skb,
+                                     &mvif->group_wcid, NULL);
        }
        spin_unlock_bh(&q->lock);
 }
 
        q = &dev->q_tx[qid];
 
        spin_lock_bh(&q->lock);
-       dev->queue_ops->tx_queue_skb(dev, q, skb, wcid, sta);
+       dev->queue_ops->tx_queue_skb(dev, qid, skb, wcid, sta);
        dev->queue_ops->kick(dev, q);
 
        if (q->queued > q->ndesc - 8 && !q->stopped) {
 {
        struct mt76_wcid *wcid = (struct mt76_wcid *) sta->drv_priv;
        struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
-       struct mt76_queue *hwq = &dev->q_tx[MT_TXQ_PSD];
 
        info->control.flags |= IEEE80211_TX_CTRL_PS_RESPONSE;
        if (last)
                               IEEE80211_TX_CTL_REQ_TX_STATUS;
 
        mt76_skb_set_moredata(skb, !last);
-       dev->queue_ops->tx_queue_skb(dev, hwq, skb, wcid, sta);
+       dev->queue_ops->tx_queue_skb(dev, MT_TXQ_PSD, skb, wcid, sta);
 }
 
 void
                    struct mt76_txq *mtxq, bool *empty)
 {
        struct ieee80211_txq *txq = mtxq_to_txq(mtxq);
+       enum mt76_txq_id qid = mt76_txq_get_qid(txq);
        struct ieee80211_tx_info *info;
        struct mt76_wcid *wcid = mtxq->wcid;
        struct sk_buff *skb;
        if (ampdu)
                mt76_check_agg_ssn(mtxq, skb);
 
-       idx = dev->queue_ops->tx_queue_skb(dev, hwq, skb, wcid, txq->sta);
+       idx = dev->queue_ops->tx_queue_skb(dev, qid, skb, wcid, txq->sta);
 
        if (idx < 0)
                return idx;
                if (cur_ampdu)
                        mt76_check_agg_ssn(mtxq, skb);
 
-               idx = dev->queue_ops->tx_queue_skb(dev, hwq, skb, wcid,
+               idx = dev->queue_ops->tx_queue_skb(dev, qid, skb, wcid,
                                                   txq->sta);
                if (idx < 0)
                        return idx;
 
 }
 
 static int
-mt76u_tx_queue_skb(struct mt76_dev *dev, struct mt76_queue *q,
+mt76u_tx_queue_skb(struct mt76_dev *dev, enum mt76_txq_id qid,
                   struct sk_buff *skb, struct mt76_wcid *wcid,
                   struct ieee80211_sta *sta)
 {
+       struct mt76_queue *q = &dev->q_tx[qid];
        struct mt76u_buf *buf;
        u16 idx = q->tail;
        int err;