diff --git a/easytier/src/proto/dns.rs b/easytier/src/proto/dns.rs
index 9fd220c3..1dee2b8d 100644
--- a/easytier/src/proto/dns.rs
+++ b/easytier/src/proto/dns.rs
@@ -1,7 +1,15 @@
 use std::fmt::Display;
+use crate::utils::DeterministicDigest;
 
 include!(concat!(env!("OUT_DIR"), "/dns.rs"));
 
+impl HeartbeatRequest {
+    pub fn update(&mut self, snapshot: DnsSnapshot) {
+        self.digest = snapshot.digest();
+        self.snapshot = Some(snapshot);
+    }
+}
+
 impl Display for ZoneData {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         writeln!(f, "; EasyTier Magic DNS zone data")?;