diff --git a/middleware/StateMachine/src/IStateMachine.cpp b/middleware/StateMachine/src/IStateMachine.cpp
index 1af3d75..e1ba6b5 100644
--- a/middleware/StateMachine/src/IStateMachine.cpp
+++ b/middleware/StateMachine/src/IStateMachine.cpp
@@ -14,7 +14,10 @@
  */
 #include "IStateMachine.h"
 #include "ILog.h"
-#include <thread>
+#include "StatusCode.h"
+#include <cstdint>
+#include <memory>
+#include <string>
 std::string State::GetStateName()
 {
     return mStateName;
diff --git a/middleware/StateMachine/src/StateMachineImpl.cpp b/middleware/StateMachine/src/StateMachineImpl.cpp
index 98a5dfe..60c4d16 100644
--- a/middleware/StateMachine/src/StateMachineImpl.cpp
+++ b/middleware/StateMachine/src/StateMachineImpl.cpp
@@ -13,7 +13,10 @@
  * limitations under the License.
  */
 #include "StateMachineImpl.h"
+#include "IStateMachine.h"
 #include "StateMachineMakePtr.h"
+#include "StatusCode.h"
+#include <memory>
 const StatusCode StateMachineImpl::CreateStateMachine(std::shared_ptr<VStateMachineHandle> &stateMachine)
 {
     return StateMachineMakePtr::GetInstance()->CreateStateMachine(stateMachine);
diff --git a/middleware/StateMachine/src/StateMachineMakePtr.cpp b/middleware/StateMachine/src/StateMachineMakePtr.cpp
index f4705dc..363f45f 100644
--- a/middleware/StateMachine/src/StateMachineMakePtr.cpp
+++ b/middleware/StateMachine/src/StateMachineMakePtr.cpp
@@ -16,8 +16,10 @@
 #include "ILog.h"
 #include "IStateMachine.h"
 #include "StateMachineImpl.h"
+#include "StatusCode.h"
 #include "state_machine.h"
-#include <thread>
+#include <memory>
+#include <utility>
 bool CreateStateMachine(void)
 {
     auto instance = std::make_shared<IStateMachine>();