diff --git a/services/flashd/BUILD.gn b/services/flashd/BUILD.gn
index 03118d99a244c24efa014ed9e45e2c781ed50396..93be03b4bd5b8303b5a4cb86b15fd2e1f20c81a9 100644
--- a/services/flashd/BUILD.gn
+++ b/services/flashd/BUILD.gn
@@ -65,6 +65,7 @@ common_sources = [
   "${HDC_PATH}/common/file_descriptor.cpp",
   "${HDC_PATH}/common/forward.cpp",
   "${HDC_PATH}/common/header.cpp",
+  "${HDC_PATH}/common/heartbeat.cpp",
   "${HDC_PATH}/common/session.cpp",
   "${HDC_PATH}/common/task.cpp",
   "${HDC_PATH}/common/tcp.cpp",
diff --git a/test/unittest/flashd_test/BUILD.gn b/test/unittest/flashd_test/BUILD.gn
index 36e84527f397be0530f644577623abe6c4a97e3d..f7c96e91f753b83532fcf38bfb002ce940b11f5e 100644
--- a/test/unittest/flashd_test/BUILD.gn
+++ b/test/unittest/flashd_test/BUILD.gn
@@ -38,6 +38,7 @@ ohos_unittest("flashd_unittest") {
     "${HDC_PATH}/common/file_descriptor.cpp",
     "${HDC_PATH}/common/forward.cpp",
     "${HDC_PATH}/common/header.cpp",
+    "${HDC_PATH}/common/heartbeat.cpp",
     "${HDC_PATH}/common/session.cpp",
     "${HDC_PATH}/common/task.cpp",
     "${HDC_PATH}/common/tcp.cpp",