diff --git a/custom_components/wellbeing/api.py b/custom_components/wellbeing/api.py
index f3ec819..2c9f877 100644
--- a/custom_components/wellbeing/api.py
+++ b/custom_components/wellbeing/api.py
@@ -201,14 +201,19 @@ def _create_entities(data):
         ]
 
         purei9_entities = [
+            ApplianceVacuum(
+                name=data.get("applianceName","Vacuum"),
+                attr="robotStatus",
+            ),
             ApplianceSensor( 
                 name="Dustbin Status",
                 attr="dustbinStatus",
                 device_class=SensorDeviceClass.ENUM,
             ),
-            ApplianceVacuum(
-                name=data.get("applianceName","Vacuum"),
+            ApplianceSensor(
+                name="Robot Status",
                 attr="robotStatus",
+                device_class=SensorDeviceClass.ENUM,
             ),
         ]