From b6e7ad3589c5d535fba687061aeea726d8466c53 Mon Sep 17 00:00:00 2001
From: Otto Winter <otto@otto-winter.com>
Date: Wed, 6 Jun 2018 11:12:14 +0200
Subject: [PATCH] Fix OTA port not being set

---
 esphomeyaml/components/ota.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/esphomeyaml/components/ota.py b/esphomeyaml/components/ota.py
index 9fb86d6cff..1dadc6a102 100644
--- a/esphomeyaml/components/ota.py
+++ b/esphomeyaml/components/ota.py
@@ -29,6 +29,8 @@ def to_code(config):
     if CONF_PASSWORD in config:
         hash_ = hashlib.md5(config[CONF_PASSWORD].encode()).hexdigest()
         add(ota.set_auth_password_hash(hash_))
+    if CONF_PORT in config:
+        add(ota.set_port(config[CONF_PORT]))
     if config[CONF_SAFE_MODE]:
         add(ota.start_safe_mode())