diff --git a/main.py b/main.py
index 284d9cd..acf6d77 100644
--- a/main.py
+++ b/main.py
@@ -178,11 +178,12 @@ if __name__ == '__main__':
             b = str(splitted_payload[2])
 
         elif topic == "server-rack-led/mode":
-            if payload == "0":
+            if payload == "Rainbow":
                 global mode
                 mode = 0
-            elif payload == "1":
+            elif payload == "Custom Color":
                 mode = 1
+                
 
     strip = Adafruit_NeoPixel(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)
     strip.begin()