diff --git a/esphome/components/api/api_connection.h b/esphome/components/api/api_connection.h
index 3b4679eb42..119cfe8084 100644
--- a/esphome/components/api/api_connection.h
+++ b/esphome/components/api/api_connection.h
@@ -8,6 +8,7 @@
 #include "api_server.h"
 #include "esphome/core/application.h"
 #include "esphome/core/component.h"
+#include "esphome/core/entity_base.h"
 
 #include <vector>