diff --git a/gaugemqttcontinuous.py b/gaugemqttcontinuous.py
index ce52b28..5e54a80 100644
--- a/gaugemqttcontinuous.py
+++ b/gaugemqttcontinuous.py
@@ -868,17 +868,14 @@ def main():
                     set_backlight_color(i, *_disco_saved_color[i], _disco_saved_brightness[i])
                 info("Disco mode ended")
             elif _disco_end_time > 0:
-                r = urandom.getrandbits(8)
-                g = urandom.getrandbits(8)
-                b = urandom.getrandbits(8)
-                r = max(r, 128)
-                g = max(g, 128)
-                b = max(b, 128)
-                for i in range(num_gauges):
-                    leds_per_gauge = BACKLIGHT_LEDS_PER_GAUGE + STATUS_LEDS_PER_GAUGE
-                    base_idx = i * leds_per_gauge
-                    for j in range(BACKLIGHT_LEDS_PER_GAUGE):
-                        leds_bl[base_idx + j] = (g, r, b)
+                for led_idx in range(total_backlight_leds):
+                    r = urandom.getrandbits(8)
+                    g = urandom.getrandbits(8)
+                    b = urandom.getrandbits(8)
+                    r = max(r, 128)
+                    g = max(g, 128)
+                    b = max(b, 128)
+                    leds_bl[led_idx] = (g, r, b)
                 leds_bl.write()
                 utime.sleep_ms(200)