diff --git a/extension/android/jni/BUCK b/extension/android/jni/BUCK
index 1b3e63002f8..1c7e714b145 100644
--- a/extension/android/jni/BUCK
+++ b/extension/android/jni/BUCK
@@ -146,6 +146,22 @@ runtime.export_file(
     name = "jni_helper.cpp",
 )
 
+runtime.export_file(
+    name = "log.cpp",
+)
+
+runtime.export_file(
+    name = "jni_helper.h",
+)
+
+runtime.export_file(
+    name = "log.h",
+)
+
+runtime.export_file(
+    name = "jni_layer_constants.h",
+)
+
 runtime.cxx_library(
     name = "jni_headers",
     exported_headers = [
diff --git a/extension/android/jni/jni_helper.cpp b/extension/android/jni/jni_helper.cpp
index f7c9c4abb74..55bca45a717 100644
--- a/extension/android/jni/jni_helper.cpp
+++ b/extension/android/jni/jni_helper.cpp
@@ -6,7 +6,7 @@
  * LICENSE file in the root directory of this source tree.
  */
 
-#include "jni_helper.h"
+#include <executorch/extension/android/jni/jni_helper.h>
 
 namespace executorch::jni_helper {
 
diff --git a/extension/android/jni/log.cpp b/extension/android/jni/log.cpp
index 663198e1271..04deb3ecfd7 100644
--- a/extension/android/jni/log.cpp
+++ b/extension/android/jni/log.cpp
@@ -6,7 +6,7 @@
  * LICENSE file in the root directory of this source tree.
  */
 
-#include "log.h"
+#include <executorch/extension/android/jni/log.h>
 
 #ifdef __ANDROID__