fix: wrong default previous menu for KlipperFlashMethodMenu

Signed-off-by: Dominik Willner <th33xitus@gmail.com>
2024-04-15 21:37:25 +02:00
parent c2b0ca5b19
commit 8de7ab7e11
1 changed files with 5 additions and 1 deletions
--- a/kiauh/components/klipper_firmware/menus/klipper_flash_menu.py
+++ b/kiauh/components/klipper_firmware/menus/klipper_flash_menu.py
@@ -53,7 +53,11 @@ class KlipperFlashMethodMenu(BaseMenu):
        self.flash_options = FlashOptions()

    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.previous_menu: Type[BaseMenu] = previous_menu
+        from core.menus.advanced_menu import AdvancedMenu
+
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else AdvancedMenu
+        )

    def set_options(self) -> None:
        self.options = {