diff --git a/main.py b/main.py
index f414ac5..3e102b0 100644
--- a/main.py
+++ b/main.py
@@ -117,15 +117,15 @@ if __name__ == '__main__':
                     blink(strip, Color(255,0,0), Color(255,255,255))
                 else:
                     print(idle_mode)
-                    if idle_mode == '1':
+                    if int(idle_mode) == '1':
                         rainbowCycle(strip)
-                    elif idle_mode == '2':
+                    elif int(idle_mode) == '2':
                         setColor(strip, Color(255,0,0))
-                    elif idle_mode == '3':
+                    elif int(idle_mode) == '3':
                         setColor(strip, Color(0,255,0))
-                    elif idle_mode == '4':
+                    elif int(idle_mode) == '4':
                         setColor(strip, Color(0,0,255))
-                    elif idle_mode == '5':
+                    elif int(idle_mode) == '5':
                         setColor(strip, Color(125,245,255))
                     else:
                         blink(strip, Color(255,0,0), Color(0,0,0))