diff --git a/waterlezer.yaml b/waterlezer.yaml
index 9080c18..c836f82 100644
--- a/waterlezer.yaml
+++ b/waterlezer.yaml
@@ -41,6 +41,9 @@ logger:
   level: INFO
 
 ota:
+  platform: esphome
+
+safe_mode:
   reboot_timeout: 10min
   num_attempts: 5