}
 
 static int
-mt7603_mcu_restart(struct mt7603_dev *dev)
+mt7603_mcu_restart(struct mt76_dev *dev)
 {
-       return __mt76_mcu_send_msg(&dev->mt76, -MCU_CMD_RESTART_DL_REQ,
+       return __mt76_mcu_send_msg(dev, -MCU_CMD_RESTART_DL_REQ,
                                   NULL, 0, true);
 }
 
 {
        static const struct mt76_mcu_ops mt7603_mcu_ops = {
                .mcu_send_msg = mt7603_mcu_msg_send,
+               .mcu_restart = mt7603_mcu_restart,
        };
 
        dev->mt76.mcu_ops = &mt7603_mcu_ops;
 
 void mt7603_mcu_exit(struct mt7603_dev *dev)
 {
-       mt7603_mcu_restart(dev);
+       __mt76_mcu_restart(&dev->mt76);
        skb_queue_purge(&dev->mt76.mmio.mcu.res_q);
 }