diff --git a/light_switch.yaml b/light_switch.yaml
index af99078..244523b 100644
--- a/light_switch.yaml
+++ b/light_switch.yaml
@@ -19,7 +19,8 @@ api:
     key: !secret ml_key
 
 ota:
-  password: !secret ml_password
+  - platform: esphome
+    password: !secret ml_password
 
 logger: