fix: error when back to main menu

Thungghuan · Oct 8, 2022 · 6d113d0 · 6d113d0
1 parent 39bfc47
commit 6d113d0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cli/main.py b/cli/main.py
@@ -69,7 +69,10 @@ def reserve_bus(self):
         reserve_menu = ReserveBusMenu(self.bus)
         result = reserve_menu.run()
 
-        if result == 0:
+        if result == 1:
+            self.change_state(MenuState.START)
+
+        else:
             self.back_main_menu()
 
     def check_reserve(self):