From d6e49bc5bcb02952102fc046e6e7f6fc0c85f45d Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Sun, 31 Oct 2021 00:27:12 +0200
Subject: [PATCH] Fix litterrobot vacuum base class (#58765)

---
 homeassistant/components/litterrobot/vacuum.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/litterrobot/vacuum.py b/homeassistant/components/litterrobot/vacuum.py
index f1717bf0209..81f9e631beb 100644
--- a/homeassistant/components/litterrobot/vacuum.py
+++ b/homeassistant/components/litterrobot/vacuum.py
@@ -17,7 +17,7 @@ from homeassistant.components.vacuum import (
     SUPPORT_STATUS,
     SUPPORT_TURN_OFF,
     SUPPORT_TURN_ON,
-    VacuumEntity,
+    StateVacuumEntity,
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import STATE_OFF
@@ -75,7 +75,7 @@ async def async_setup_entry(
     )
 
 
-class LitterRobotCleaner(LitterRobotControlEntity, VacuumEntity):
+class LitterRobotCleaner(LitterRobotControlEntity, StateVacuumEntity):
     """Litter-Robot "Vacuum" Cleaner."""
 
     @property
@@ -148,4 +148,5 @@ class LitterRobotCleaner(LitterRobotControlEntity, VacuumEntity):
             "power_status": self.robot.power_status,
             "status_code": self.robot.status_code,
             "last_seen": self.robot.last_seen,
+            "status": self.status,
         }