diff --git a/localbytes-plug-pm.yaml b/localbytes-plug-pm.yaml
index c590012..e5a3810 100644
--- a/localbytes-plug-pm.yaml
+++ b/localbytes-plug-pm.yaml
@@ -24,7 +24,7 @@ esphome:
   # configuration and updates.
   project:
     name: localbytes.plug-pm
-    version: "1.6.1"
+    version: "1.6.2"
 
 esp8266:
   board: esp01_1m
@@ -155,7 +155,7 @@ switch:
     name: "Disable LED"
     id: disable_led
     icon: "mdi:led-variant-off"
-    restore_state: true
+    restore_mode: RESTORE_DEFAULT_OFF
     optimistic: true
     on_turn_on:
       #Flash twice
@@ -191,8 +191,8 @@ switch:
   - platform: template
     name: "Disable Button"
     id: disable_button
-    icon: "mdi:toggle-switch-off-outline"
-    restore_state: true
+    icon: "mdi:toggle-switch-off-outline"    
+    restore_mode: RESTORE_DEFAULT_OFF
     optimistic: true
     on_turn_on:
       #Flash thrice