diff --git a/include/aidge/utils/Attributes.hpp b/include/aidge/utils/Attributes.hpp
index 76875f15ff4229522e6208b0edb23ec519ff59ce..d3444000191022b575adaf1430319479daa5d4fc 100644
--- a/include/aidge/utils/Attributes.hpp
+++ b/include/aidge/utils/Attributes.hpp
@@ -18,6 +18,7 @@
 #endif
 #include <vector>
 #include <string>
+#include <set>
 
 #ifdef PYBIND
 namespace py = pybind11;
diff --git a/include/aidge/utils/Utils.hpp b/include/aidge/utils/Utils.hpp
index 7c0c03c82ff252b6175d3c9bbe5395bb05127c9f..71817dcfc9713ad36a74175affd21b03cb6ed181 100644
--- a/include/aidge/utils/Utils.hpp
+++ b/include/aidge/utils/Utils.hpp
@@ -14,6 +14,7 @@
 #define AIDGE_UTILS_H_
 
 #include <cstdio>
+#include <memory>
 
 #ifdef NO_EXCEPTIONS
 #define AIDGE_THROW_OR_ABORT(ex, ...) \