diff --git a/custom_components/localtuya/vacuum.py b/custom_components/localtuya/vacuum.py
index 15a1d6b..9a14399 100644
--- a/custom_components/localtuya/vacuum.py
+++ b/custom_components/localtuya/vacuum.py
@@ -154,7 +154,7 @@ class LocaltuyaVacuum(LocalTuyaEntity, StateVacuumEntity):
         return self._battery_level
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         """Return the specific state attributes of this vacuum cleaner."""
         return self._attrs