diff --git a/dnnlibrary/src/Model.cpp b/dnnlibrary/src/Model.cpp
index e785dcc..bc9cdc0 100644
--- a/dnnlibrary/src/Model.cpp
+++ b/dnnlibrary/src/Model.cpp
@@ -11,6 +11,7 @@
 
 #include <common/helper.h>
 #include <glog/logging.h>
+#include <nnapi_helper.h>
 
 namespace dnn {
 template void Model::Predict<float>(const std::vector<float> &);
diff --git a/dnnlibrary/src/ModelBuilder.cpp b/dnnlibrary/src/ModelBuilder.cpp
index da0b90c..2f05dd3 100644
--- a/dnnlibrary/src/ModelBuilder.cpp
+++ b/dnnlibrary/src/ModelBuilder.cpp
@@ -15,6 +15,7 @@
 #include <common/data_types.h>
 #include <common/helper.h>
 #include <glog/logging.h>
+#include <nnapi_helper.h>
 #include <operand_helper.h>
 #include "android_log_helper.h"