From 42a2e704c1f4b3a67993009129271c9e243277ce Mon Sep 17 00:00:00 2001
From: Thomas Deblock <deblock.thomas.62@gmail.com>
Date: Sat, 23 Sep 2023 16:32:52 +0200
Subject: [PATCH] fix: use StateVacuumEntity instead of VacuumEntity

---
 custom_components/proscenic/vacuum.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/custom_components/proscenic/vacuum.py b/custom_components/proscenic/vacuum.py
index fcf1783..52a50cc 100755
--- 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):