diff --git a/sonoff/xdrv_16_tuyadimmer.ino b/sonoff/xdrv_16_tuyadimmer.ino
index 65d316ca8..218ef98fd 100644
--- a/sonoff/xdrv_16_tuyadimmer.ino
+++ b/sonoff/xdrv_16_tuyadimmer.ino
@@ -235,12 +235,14 @@ void TuyaSetWifiLed(){
       case WIFI_WPSCONFIG:
         wifi_state = 0x01;
         break;
-      case WIFI_WAIT:
-      case WIFI_RETRY:
-        wifi_state = 0x02;
       case WIFI_RESTART:
         wifi_state =  0x03;
         break;
+      case WIFI_WAIT:
+      case WIFI_RETRY:
+      default:
+        wifi_state = 0x02;
+        break;
     }
 
     snprintf_P(log_data, sizeof(log_data), "TYA: Set WiFi LED to state %d (%d)", WifiState(), wifi_state);