diff --git a/esphome/codegen.py b/esphome/codegen.py
index 8e02ec1164..6decd77c62 100644
--- a/esphome/codegen.py
+++ b/esphome/codegen.py
@@ -12,6 +12,7 @@ from esphome.cpp_generator import (  # noqa: F401
     ArrayInitializer,
     Expression,
     LineComment,
+    LogStringLiteral,
     MockObj,
     MockObjClass,
     Pvariable,