From f1d8af8e3d37e822102e7bfd819219ea497cc988 Mon Sep 17 00:00:00 2001
From: Blizzard Finnegan <blizzardfinnegan@gmail.com>
Date: Mon, 19 Jun 2023 14:58:15 -0400
Subject: [PATCH] Add end handling

---
 src/device.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/device.rs b/src/device.rs
index 5cc49a7..a0b2f78 100755
--- a/src/device.rs
+++ b/src/device.rs
@@ -373,6 +373,8 @@ impl Device{
             }
         }
         self.usb_tty.write_to_device(Command::DebugMenu);
+        while self.usb_tty.read_from_device(None) != Response::DebugMenu {}
+        self.current_state = State::DebugMenu;
         self.reboot();
         self.load_values();
         self.save_values();