diff --git a/Android.bp b/Android.bp
index c78aaf5..8496784 100644
--- a/Android.bp
+++ b/Android.bp
@@ -16,6 +16,11 @@
     name: "libtextclassifier_hash_headers",
     vendor_available: true,
     export_include_dirs: ["."],
+    apex_available: [
+        "//apex_available:platform",
+        "com.android.neuralnetworks",
+        "test_com.android.neuralnetworks",
+    ],
 }
 
 cc_defaults {
@@ -45,6 +50,11 @@
     vendor_available: true,
     sdk_version: "current",
     stl: "libc++_static",
+    apex_available: [
+        "//apex_available:platform",
+        "com.android.neuralnetworks",
+        "test_com.android.neuralnetworks",
+    ],
 }
 
 java_library_static {