diff --git a/src/utils/Decompressor.h b/src/utils/Decompressor.h
index 4e3c298..5551b57 100644
--- a/src/utils/Decompressor.h
+++ b/src/utils/Decompressor.h
@@ -4,6 +4,7 @@
 #include <cstdint>
 #include <unordered_map>
 #include <yaml-cpp/yaml.h>
+#include <optional>
 
 enum class CompressionType {
     MIO0,
diff --git a/src/utils/TorchUtils.h b/src/utils/TorchUtils.h
index 330c7fa..e3d89e6 100644
--- a/src/utils/TorchUtils.h
+++ b/src/utils/TorchUtils.h
@@ -3,6 +3,7 @@
 #include <string>
 #include <fstream>
 #include <sstream>
+#include <iomanip>
 
 namespace Torch {
 template< typename T >