diff --git a/custom_components/ecovacs/manifest.json b/custom_components/ecovacs/manifest.json
index 3e7121d..e58d1b2 100644
--- a/custom_components/ecovacs/manifest.json
+++ b/custom_components/ecovacs/manifest.json
@@ -1,7 +1,7 @@
 {
   "domain": "ecovacs",
   "name": "Ecovacs Bumper",
-  "version": "1.4.4",
+  "version": "1.4.5",
   "documentation": "https://github.com/bittles/ha_ecovacs_bumper",
   "issue_tracker": "https://github.com/bittles/ha_ecovacs_bumper/issues",
   "requirements": ["sleekxmppfs==1.4.1", "requests>=2.18", "pycryptodome>=3.4", "pycountry-convert>=0.5", "paho-mqtt>=1.4", "stringcase>=1.2"],
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