diff --git a/posts/TensorRT-introduction/Makefile b/posts/TensorRT-introduction/Makefile
index cc3ef8e..016f6d1 100644
--- a/posts/TensorRT-introduction/Makefile
+++ b/posts/TensorRT-introduction/Makefile
@@ -25,7 +25,7 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 CUDA_INSTALL_DIR=/usr/local/cuda
 
-CXXFLAGS=-std=c++11 -Wall -I$(CUDA_INSTALL_DIR)/include
+CXXFLAGS=-std=c++11 -Wall -I$(CUDA_INSTALL_DIR)/include -DONNX_ML=1
 LDFLAGS=-L$(CUDA_INSTALL_DIR)/lib64 -L$(CUDA_INSTALL_DIR)/lib64/stubs 
 LDLIBS=-Wl,--start-group -lnvinfer -lnvonnxparser -lcudart_static -lrt -ldl -lpthread -lonnx -lonnx_proto -lprotobuf -lstdc++ -lm -Wl,--end-group
 
diff --git a/posts/TensorRT-introduction/simpleOnnx.cpp b/posts/TensorRT-introduction/simpleOnnx.cpp
index ea41b7f..53b1dea 100644
--- a/posts/TensorRT-introduction/simpleOnnx.cpp
+++ b/posts/TensorRT-introduction/simpleOnnx.cpp
@@ -34,6 +34,8 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <cmath>
+#include <numeric>
 
 using namespace nvinfer1;
 using namespace std;
diff --git a/posts/TensorRT-introduction/simpleOnnx_1.cpp b/posts/TensorRT-introduction/simpleOnnx_1.cpp
index a2e66d5..90d97ce 100644
--- a/posts/TensorRT-introduction/simpleOnnx_1.cpp
+++ b/posts/TensorRT-introduction/simpleOnnx_1.cpp
@@ -34,6 +34,8 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <cmath>
+#include <numeric>
 
 using namespace nvinfer1;
 using namespace std;
diff --git a/posts/TensorRT-introduction/simpleOnnx_2.cpp b/posts/TensorRT-introduction/simpleOnnx_2.cpp
index 3432436..2b239aa 100644
--- a/posts/TensorRT-introduction/simpleOnnx_2.cpp
+++ b/posts/TensorRT-introduction/simpleOnnx_2.cpp
@@ -34,6 +34,8 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <cmath>
+#include <numeric>
 
 using namespace nvinfer1;
 using namespace std;