diff --git a/miio/vacuum.py b/miio/vacuum.py
index 011586f72..e00bfae48 100644
--- a/miio/vacuum.py
+++ b/miio/vacuum.py
@@ -129,6 +129,8 @@ def resume_or_start(self):
         status = self.status()
         if status.in_zone_cleaning and (status.is_paused or status.got_error):
             return self.resume_zoned_clean()
+        if status.in_segment_cleaning and (status.is_paused or status.got_error):
+            return self.resume_segment_clean()
 
         return self.start()