diff --git a/pymodoro.py b/pymodoro.py index 71e50f6..3e259b4 100644 --- a/pymodoro.py +++ b/pymodoro.py @@ -31,6 +31,9 @@ class Pymodoro: def on_btn(self, snapshot, timestamp): if self.phase == PH_BREAK_OVER: self.switch(PH_WORK) + + elif self.phase == PH_WORK: + self.switch(PH_WORK) # restart elif self.phase == PH_WORK_OVER: self.switch(PH_BREAK)