diff --git a/CAPI/cpp/API/include/state.h b/CAPI/cpp/API/include/state.h
index 1440a2d..02f5543 100644
--- a/CAPI/cpp/API/include/state.h
+++ b/CAPI/cpp/API/include/state.h
@@ -5,6 +5,7 @@
 #include <vector>
 #include <array>
 #include <map>
+#include <memory>
 
 #include "structures.h"
 
diff --git a/CAPI/cpp/API/include/structures.h b/CAPI/cpp/API/include/structures.h
index 613e04e..1b10d20 100644
--- a/CAPI/cpp/API/include/structures.h
+++ b/CAPI/cpp/API/include/structures.h
@@ -5,6 +5,8 @@
 #include <cstdint>
 #include <array>
 #include <map>
+#include <vector>
+#include <string>
 
 namespace THUAI6
 {
diff --git a/CAPI/cpp/API/include/utils.hpp b/CAPI/cpp/API/include/utils.hpp
index 3472cf6..3c61f8a 100644
--- a/CAPI/cpp/API/include/utils.hpp
+++ b/CAPI/cpp/API/include/utils.hpp
@@ -5,6 +5,8 @@
 
 #include <cstdint>
 #include <cmath>
+#include <map>
+#include <vector>
 #include "Message2Clients.pb.h"
 #include "Message2Server.pb.h"
 #include "MessageType.pb.h"