fix: use StateVacuumEntity instead of VacuumEntity

deblockt · Sep 23, 2023 · 42a2e70 · 42a2e70
1 parent af9997d
commit 42a2e70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/proscenic/vacuum.py b/custom_components/proscenic/vacuum.py
@@ -18,7 +18,7 @@
     STATE_IDLE,
     STATE_PAUSED,
     STATE_RETURNING,
-    VacuumEntity,
+    StateVacuumEntity,
     PLATFORM_SCHEMA
 )
 from homeassistant.const import (
@@ -102,7 +102,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
     async_add_entities(vacuums, update_before_add = False)
 
 
-class ProscenicVacuum(VacuumEntity):
+class ProscenicVacuum(StateVacuumEntity):
     """790T Vacuums such as Deebot."""
 
     def __init__(self, device, name):