struct mt7615_rate_desc rd;
        u32 w5, w27, addr;
 
-       if (mt76_is_usb(&dev->mt76)) {
+       if (!mt76_is_mmio(&dev->mt76)) {
                mt7615_mac_queue_rate_update(phy, sta, probe_rate, rates);
                return;
        }
 
        mt76_wcid_key_setup(&dev->mt76, wcid,
                            cmd == SET_KEY ? key : NULL);
 
-       if (mt76_is_usb(&dev->mt76))
-               err = mt7615_queue_key_update(dev, cmd, msta, key);
-       else
+       if (mt76_is_mmio(&dev->mt76))
                err = mt7615_mac_wtbl_set_key(dev, wcid, key, cmd);
+       else
+               err = mt7615_queue_key_update(dev, cmd, msta, key);
 
        mt7615_mutex_release(dev);