diff --git a/src/input_handler.h b/src/input_handler.h
index 2b2abc0..ca90eca 100644
--- a/src/input_handler.h
+++ b/src/input_handler.h
@@ -2,6 +2,7 @@
 
 #include <string>
 #include <functional>
+#include <memory>
 
 enum class InputMode {
     NORMAL,    // 正常浏览模式
diff --git a/src/text_renderer.h b/src/text_renderer.h
index 3737a5c..8298f0a 100644
--- a/src/text_renderer.h
+++ b/src/text_renderer.h
@@ -3,6 +3,7 @@
 #include "html_parser.h"
 #include <string>
 #include <vector>
+#include <memory>
 #include <curses.h>
 
 // 渲染后的行信息