From 5c335590960d5c1f4b3e012683d388800aa094d0 Mon Sep 17 00:00:00 2001
From: Naomi Pentrel <5212232+npentrel@users.noreply.github.com>
Date: Sat, 22 Jun 2024 22:37:36 +0100
Subject: [PATCH] Call cleanup before handlers for menu

---
 modules/app_components/menu.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/app_components/menu.py b/modules/app_components/menu.py
index 05ccefa..b5e4216 100644
--- a/modules/app_components/menu.py
+++ b/modules/app_components/menu.py
@@ -62,9 +62,11 @@ def _handle_buttondown(self, event: ButtonDownEvent):
                     self.menu_items[self.position % len(self.menu_items)]
                 )
         if BUTTON_TYPES["CANCEL"] in event.button:
+            self._cleanup()
             if self.back_handler is not None:
                 self.back_handler()
         if BUTTON_TYPES["CONFIRM"] in event.button:
+            self._cleanup()
             if self.select_handler is not None:
                 self.select_handler(
                     self.menu_items[self.position % len(self.menu_items)],