diff --git a/visualdl/logic/im.h b/visualdl/logic/im.h
index 6d6ab428db2878e095868a203d6a73318c067794..f32cbc217f7e5d9520d72923e419433df59a787f 100644
--- a/visualdl/logic/im.h
+++ b/visualdl/logic/im.h
@@ -3,6 +3,7 @@
 
 #include <glog/logging.h>
 #include <memory>
+#include <mutex>
 #include <string>
 
 #include "visualdl/storage/storage.h"
diff --git a/visualdl/storage/storage.h b/visualdl/storage/storage.h
index d3c7ca5bad05c45146a21cad32d606c3d3ccdbc8..31b9ab4ca6ac0e0e81c164e89313769e602751a0 100644
--- a/visualdl/storage/storage.h
+++ b/visualdl/storage/storage.h
@@ -4,6 +4,7 @@
 #include <time.h>
 #include <map>
 #include <memory>
+#include <mutex>
 #include <string>
 
 #include "visualdl/storage/storage.pb.h"