diff --git a/custom_components/xiaomi_home/vacuum.py b/custom_components/xiaomi_home/vacuum.py
index c02f925..7a61596 100644
--- a/custom_components/xiaomi_home/vacuum.py
+++ b/custom_components/xiaomi_home/vacuum.py
@@ -202,12 +202,14 @@ class Vacuum(MIoTServiceEntity, StateVacuumEntity):
             elif action.name == 'identify':
                 self._attr_supported_features |= VacuumEntityFeature.LOCATE
                 self._action_identify = action
-        
-        # Use start-charge from battery service as fallback if stop-and-gocharge is not available
+
+        # Use start-charge from battery service as fallback
+        # if stop-and-gocharge is not available
         if self._action_stop_and_gocharge is None:
             for action in entity_data.actions:
                 if action.name == 'start-charge':
-                    self._attr_supported_features |= VacuumEntityFeature.RETURN_HOME
+                    self._attr_supported_features |= (
+                        VacuumEntityFeature.RETURN_HOME)
                     self._action_stop_and_gocharge = action
                     break