diff --git a/visualdl/storage/storage.h b/visualdl/storage/storage.h
index 6c60b9302ddfd1e7409a696fc496dfdae6d3c574..d3c7ca5bad05c45146a21cad32d606c3d3ccdbc8 100644
--- a/visualdl/storage/storage.h
+++ b/visualdl/storage/storage.h
@@ -3,6 +3,7 @@
 
 #include <time.h>
 #include <map>
+#include <memory>
 #include <string>
 
 #include "visualdl/storage/storage.pb.h"