diff --git a/services/distributeddataservice/service/test/BUILD.gn b/services/distributeddataservice/service/test/BUILD.gn
index a5f539cc583e09e13683df98b3177ccce398fbce..57b0a71f0799f4da10b8de74e4d9de1a234dcfc2 100644
--- a/services/distributeddataservice/service/test/BUILD.gn
+++ b/services/distributeddataservice/service/test/BUILD.gn
@@ -906,6 +906,7 @@ ohos_unittest("UdmfRunTimeStoreTest") {
     "image_framework:image",
     "ipc:ipc_core",
     "kv_store:distributeddata_inner",
+    "kv_store:datamgr_common",
     "kv_store:distributeddb",
     "openssl:libcrypto_shared",
     "samgr:samgr_proxy",
@@ -1384,6 +1385,7 @@ ohos_unittest("UdmfPreProcessUtilsMockTest") {
     "hisysevent:libhisysevent",
     "hitrace:hitrace_meter",
     "hitrace:libhitracechain",
+    "kv_store:datamgr_common",
     "udmf:udmf_client",
   ]
 }
diff --git a/services/distributeddataservice/service/udmf/BUILD.gn b/services/distributeddataservice/service/udmf/BUILD.gn
index 8176548a1cdd0cdfd3b02c6a6a9689f315786b0e..8373666559f531fa05070f73b0ef145094abfcec 100644
--- a/services/distributeddataservice/service/udmf/BUILD.gn
+++ b/services/distributeddataservice/service/udmf/BUILD.gn
@@ -28,9 +28,6 @@ config("module_public_config") {
     "${data_service_path}/service/udmf/utd",
     "${data_service_path}/service/udmf",
     "${data_service_path}/service/bootstrap/include",
-    "${kv_store_path}/interfaces/innerkits/distributeddata/include",
-    "${kv_store_path}/framework/libs/distributeddb/interfaces/include",
-    "${kv_store_common_path}",
     "${udmf_path}/framework/common",
     "${udmf_path}/interfaces/innerkits/common",
     "${udmf_path}/interfaces/innerkits/data",
@@ -87,6 +84,7 @@ ohos_source_set("udmf_server") {
     "hitrace:hitrace_meter",
     "hitrace:libhitracechain",
     "ipc:ipc_core",
+    "kv_store:datamgr_common",
     "kv_store:distributeddb",
     "udmf:udmf_client",
     "udmf:utd_client",