diff --git a/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp b/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp
index 46f10bde1fbc64efbeb4c677283c536f4869779e..50aa8c78441eab5a666707f57c3b1110d77be6ff 100755
--- a/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp
+++ b/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp
@@ -2,7 +2,6 @@
 #define BATCH_STREAM_H
 
 #include "NvInfer.h"
-#include "dnn_utils.hpp"
 #include <algorithm>
 #include <assert.h>
 #include <stdio.h>
@@ -67,8 +66,7 @@ public:
         }
 
         int x = _batchCount;
-        for (int i = 0; i < skipCount; i++)
-            next();
+        for (std::size_t i = 0; i < skipCount; ++i) next();
         _batchCount = x;
     }
 
diff --git a/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp b/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp
index 61308b99b5ec077b9685e95ceb1ec14eb31b63a4..601692528504b03f7451b7cd71f2f1594a218e8a 100644
--- a/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp
+++ b/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp
@@ -1,6 +1,9 @@
 #include "Graph.hpp"
 #include <fstream>
 #include <sstream>
+#include "BatchStream.hpp"
+#include "IInt8EntropyCalibrator.hpp"
+#include <dirent.h>
 
 Graph::Graph(std::string const& filePath,
              unsigned int device_id,