.tx_complete_skb = mt76x02u_tx_complete_skb,
                .tx_status_data = mt76x02_tx_status_data,
                .rx_skb = mt76x02_queue_rx_skb,
+               .sta_ps = mt76x02_sta_ps,
                .sta_add = mt76x02_sta_add,
                .sta_remove = mt76x02_sta_remove,
        };
 
        int idx = msta->wcid.idx;
 
        mt76_stop_tx_queues(&dev->mt76, sta, true);
-       mt76x02_mac_wcid_set_drop(dev, idx, ps);
+       if (mt76_is_mmio(dev))
+               mt76x02_mac_wcid_set_drop(dev, idx, ps);
 }
 EXPORT_SYMBOL_GPL(mt76x02_sta_ps);
 
 
                .tx_complete_skb = mt76x02u_tx_complete_skb,
                .tx_status_data = mt76x02_tx_status_data,
                .rx_skb = mt76x02_queue_rx_skb,
+               .sta_ps = mt76x02_sta_ps,
                .sta_add = mt76x02_sta_add,
                .sta_remove = mt76x02_sta_remove,
        };