diff --git a/roborock/devices/traits/a01/__init__.py b/roborock/devices/traits/a01/__init__.py
index 537c84377..fd2045f24 100644
--- a/roborock/devices/traits/a01/__init__.py
+++ b/roborock/devices/traits/a01/__init__.py
@@ -53,7 +53,7 @@
 from roborock.devices.transport.mqtt_channel import MqttChannel
 from roborock.roborock_message import RoborockDyadDataProtocol, RoborockZeoProtocol
 
-__init__ = [
+__all__ = [
     "DyadApi",
     "ZeoApi",
 ]