diff --git a/src/utils/Decompressor.h b/src/utils/Decompressor.h
index 5551b57..5d321e5 100644
--- a/src/utils/Decompressor.h
+++ b/src/utils/Decompressor.h
@@ -5,6 +5,7 @@
 #include <unordered_map>
 #include <yaml-cpp/yaml.h>
 #include <optional>
+#include <cstdint>
 
 enum class CompressionType {
     MIO0,
diff --git a/src/utils/TorchUtils.h b/src/utils/TorchUtils.h
index 6c8b6c4..36ce46d 100644
--- a/src/utils/TorchUtils.h
+++ b/src/utils/TorchUtils.h
@@ -5,6 +5,7 @@
 #include <sstream>
 #include <iomanip>
 #include <algorithm>
+#include <cstdint>
 
 namespace Torch {
 template< typename T >