1 files changed, 5 insertions, 0 deletions

diff --git a/tv_light/tv_light.ino b/tv_light/tv_light.ino
index 5588cef..655be24 100644
--- a/tv_light/tv_light.ino
+++ b/tv_light/tv_light.ino
@@ -132,6 +132,8 @@ void before()
 #else
       uint8_t level = sensor->dimmer.level;
 #endif
+      if ((sensor->type & SENSOR_RELAY) && !relayRead(sensor))
+        level = 0;
       analogWrite(sensor->dimmer.pin, pwmValue(level));
     }
   }
@@ -268,6 +270,9 @@ void relayWrite(struct sensor_t *sensor, bool state, bool send_update)
 
   digitalWrite(sensor->relay.pin, state ? RELAY_ON : RELAY_OFF);
 
+  if (sensor->type & SENSOR_DIMMER)
+    analogWrite(sensor->dimmer.pin, state ? pwmValue(sensor->dimmer.level) : 0);
+
 #ifdef SAVE_RESTORE
   saveState(sensor->id, state ? RELAY_ON : RELAY_OFF);
 #endif