diff --git a/mlmodel/src/MILBlob/Fp8.cpp b/mlmodel/src/MILBlob/Fp8.cpp
index d21f17729..2176fad97 100644
--- a/mlmodel/src/MILBlob/Fp8.cpp
+++ b/mlmodel/src/MILBlob/Fp8.cpp
@@ -6,6 +6,7 @@
 #include "MILBlob/Fp8.hpp"
 
 #include <cmath>
+#include <stdexcept>
 
 using namespace MILBlob;
 
diff --git a/modelpackage/src/utils/JsonMap.hpp b/modelpackage/src/utils/JsonMap.hpp
index 0d7dc3f4e..962d25b61 100644
--- a/modelpackage/src/utils/JsonMap.hpp
+++ b/modelpackage/src/utils/JsonMap.hpp
@@ -10,6 +10,7 @@
 #include <iostream>
 #include <vector>
 #include <string>
+#include <memory>
 
 class JsonMapImpl;