diff --git a/automation.h b/automation.h
index 39060b3..ccaab7b 100644
--- a/automation.h
+++ b/automation.h
@@ -1,6 +1,6 @@
 #pragma once
 
-#include "uart.h"
+#include "marlin2.h"
 #include "esphome/core/automation.h"
 
 #include <vector>