diff --git a/esphome/components/esphome/ota/__init__.py b/esphome/components/esphome/ota/__init__.py
index 901657ec82..bf5c438f9b 100644
--- a/esphome/components/esphome/ota/__init__.py
+++ b/esphome/components/esphome/ota/__init__.py
@@ -1,7 +1,8 @@
 import logging
 
 import esphome.codegen as cg
-from esphome.components.ota import BASE_OTA_SCHEMA, OTAComponent, ota_to_code
+from esphome.components.ota import BASE_OTA_SCHEMA, ota_to_code
+from esphome.components.ota_base import OTAComponent
 from esphome.config_helpers import merge_config
 import esphome.config_validation as cv
 from esphome.const import (