From b5313e5c820634a0737ac9008e37b9f4db009035 Mon Sep 17 00:00:00 2001
From: Rob Pilling <robpilling@gmail.com>
Date: Wed, 1 May 2024 22:19:37 +0100
Subject: [PATCH] promenu: pass `back` into setUI() for the back button

---
 apps/promenu/bootb2.ts | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/apps/promenu/bootb2.ts b/apps/promenu/bootb2.ts
index 8abbd1dc2..ceedc24ce 100644
--- a/apps/promenu/bootb2.ts
+++ b/apps/promenu/bootb2.ts
@@ -163,11 +163,6 @@ type ActualMenuItem = Exclude<Menu["..."], MenuOptions | undefined>;
 
   l.draw();
 
-  Bangle.setUI("updown", dir => {
-    if (dir) l.move(dir);
-    else l.select();
-  });
-
   let back = options.back;
   if (!back) {
     const backItem = items["< Back"];
@@ -183,5 +178,16 @@ type ActualMenuItem = Exclude<Menu["..."], MenuOptions | undefined>;
     })
   }
 
+  Bangle.setUI(
+      {
+          mode: "updown",
+          back,
+      },
+      dir => {
+          if (dir) l.move(dir);
+          else l.select();
+      }
+  );
+
   return l;
 };