diff --git a/roborock/devices/traits/b01/q7/__init__.py b/roborock/devices/traits/b01/q7/__init__.py
index 0e1ed93a..2bfa0a6d 100644
--- a/roborock/devices/traits/b01/q7/__init__.py
+++ b/roborock/devices/traits/b01/q7/__init__.py
@@ -10,7 +10,7 @@
     SCWindMapping,
     WaterLevelMapping,
 )
-from roborock.devices.b01_channel import send_decoded_command
+from roborock.devices.b01_channel import CommandType, ParamsType, send_decoded_command
 from roborock.devices.mqtt_channel import MqttChannel
 from roborock.devices.traits import Trait
 from roborock.roborock_message import RoborockB01Props
@@ -100,7 +100,7 @@ async def find_me(self) -> None:
             params={},
         )
 
-    async def send(self, command: RoborockB01Q7Methods, params: dict) -> Any:
+    async def send(self, command: CommandType, params: ParamsType) -> Any:
         """Send a command to the device."""
         return await send_decoded_command(
             self._channel,