diff --git a/configuration/esphome/usb-switch.yaml b/configuration/esphome/usb-switch.yaml
index fcecec1..9ea6a81 100644
--- a/configuration/esphome/usb-switch.yaml
+++ b/configuration/esphome/usb-switch.yaml
@@ -1,7 +1,9 @@
 # Sinilink USB Switch Module (XY-WFUSB)
 esphome:
   name: usb-switch
-  platform: ESP8266
+
+esp8266:
+  restore_from_flash: true
   board: esp01_1m
 
 
@@ -72,4 +74,4 @@ binary_sensor:
       mode: INPUT_PULLUP
       inverted: True
     on_press:
-      - switch.toggle: "usb_switch_relay"
\ No newline at end of file
+      - switch.toggle: "usb_switch_relay"