diff --git a/ai/neural_network_runtime/BUILD.gn b/ai/neural_network_runtime/BUILD.gn
index 66b4fa143d26723c5859c3d30f14adc5965bbaff..25323444ca49b4df78a5e0a76d807100a10a71b2 100644
--- a/ai/neural_network_runtime/BUILD.gn
+++ b/ai/neural_network_runtime/BUILD.gn
@@ -20,8 +20,8 @@ ohos_ndk_library("libneural_network_core_ndk") {
   min_compact_version = "11"
   system_capability = "SystemCapability.AI.NeuralNetworkRuntime"
   system_capability_headers = [
-    "neural_network_core.h",
-    "neural_network_runtime_type.h",
+    "neural_network_runtime/neural_network_core.h",
+    "neural_network_runtime/neural_network_runtime_type.h",
   ]
 }
 
@@ -32,8 +32,8 @@ ohos_ndk_library("libneural_network_runtime_ndk") {
   min_compact_version = "9"
   system_capability = "SystemCapability.AI.NeuralNetworkRuntime"
   system_capability_headers = [
-    "neural_network_runtime.h",
-    "neural_network_runtime_type.h",
+    "neural_network_runtime/neural_network_runtime.h",
+    "neural_network_runtime/neural_network_runtime_type.h",
   ]
 }