From e33111d38f729b47e03e9699eeb5b73637a92496 Mon Sep 17 00:00:00 2001
From: Selene ToyKeeper
Date: Thu, 30 Nov 2023 13:37:23 -0700
Subject: fixed RGB aux turning on in momentary mode when post-off voltage was
 enabled

---
 ui/anduril/momentary-mode.c | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

(limited to 'ui')

diff --git a/ui/anduril/momentary-mode.c b/ui/anduril/momentary-mode.c
index 94f6bc2..327302a 100644
--- a/ui/anduril/momentary-mode.c
+++ b/ui/anduril/momentary-mode.c
@@ -47,14 +47,17 @@ uint8_t momentary_state(Event event, uint16_t arg) {
         }
         else {
         #endif
+            // turn off lighted button
+            #ifdef USE_INDICATOR_LED
+                indicator_led(0);
+            #elif defined(USE_AUX_RGB_LEDS)
+                rgb_led_set(0);
+            #endif
+            #ifdef USE_BUTTON_LED
+                button_led_set(0);
+            #endif
             if (arg > TICKS_PER_SECOND*5) {  // sleep after 5 seconds
                 go_to_standby = 1;  // sleep while light is off
-                // turn off lighted button
-                #ifdef USE_INDICATOR_LED
-                indicator_led(0);
-                #elif defined(USE_AUX_RGB_LEDS)
-                rgb_led_update(0, 0);
-                #endif
             }
         #ifdef USE_STROBE_STATE
         }
-- 
cgit v1.2.3