diff --git a/custom_components/ecovacs/sucks.py b/custom_components/ecovacs/sucks.py
index 264ac1e..8ee481b 100644
--- a/custom_components/ecovacs/sucks.py
+++ b/custom_components/ecovacs/sucks.py
@@ -7,6 +7,7 @@ import os
 from sleekxmppfs.xmlstream import ET
 from sleekxmppfs.exceptions import XMPPError
 
+from . import sucks_api
 from .sucks_mqtt import EcoVacsIOTMQ
 from .sucks_xmpp import EcoVacsXMPP