diff --git a/Android.bp b/Android.bp
index 8496784..52599a7 100644
--- a/Android.bp
+++ b/Android.bp
@@ -119,6 +119,7 @@
     ],
 
     header_libs: [
+        "jni_headers",
         "tensorflow_headers",
         "flatbuffer_headers",
     ],