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..c4ec46a9dfabeda893f4e2a48e92fe9cd2c576fa 100644
--- a/test/unittest/flashd_test/BUILD.gn
+++ b/test/unittest/flashd_test/BUILD.gn
@@ -44,6 +44,7 @@ ohos_unittest("flashd_unittest") {
     "${HDC_PATH}/common/transfer.cpp",
     "${HDC_PATH}/common/usb.cpp",
     "${HDC_PATH}/common/uv_status.cpp",
+    "${HDC_PATH}/common/heartbeat.cpp",
     "${HDC_PATH}/daemon/daemon_app.cpp",
     "${HDC_PATH}/daemon/daemon_forward.cpp",
     "${HDC_PATH}/daemon/daemon_tcp.cpp",