diff --git a/custom_components/ecovacs/sucks_mqtt.py b/custom_components/ecovacs/sucks_mqtt.py
index e4a97bb..3a04897 100644
--- a/custom_components/ecovacs/sucks_mqtt.py
+++ b/custom_components/ecovacs/sucks_mqtt.py
@@ -9,6 +9,7 @@ from paho.mqtt.client import Client  as ClientMQTT
 from paho.mqtt import publish as MQTTPublish
 from paho.mqtt import subscribe as MQTTSubscribe
 from sleekxmppfs.xmlstream import ET
+from .sucks import EcoVacsAPI
 
 from .const import LOGGER