diff --git a/miio/viomivacuum.py b/miio/viomivacuum.py index 6c204baca..897cb733b 100644 --- a/miio/viomivacuum.py +++ b/miio/viomivacuum.py @@ -56,7 +56,7 @@ class ViomiVacuumState(Enum): Cleaning = 3 Returning = 4 Docked = 5 - Mopping = 6 + VacuumingAndMopping = 6 class ViomiMode(Enum): @@ -129,7 +129,10 @@ def state(self): @property def is_on(self) -> bool: """True if cleaning.""" - cleaning_states = [ViomiVacuumState.Cleaning, ViomiVacuumState.Mopping] + cleaning_states = [ + ViomiVacuumState.Cleaning, + ViomiVacuumState.VacuumingAndMopping, + ] return self.state in cleaning_states @property