diff --git a/Cargo.lock b/Cargo.lock
index 4ebb1da78..5e2a78087 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2567,6 +2567,7 @@ dependencies = [
  "cfg-if",
  "clap",
  "devices",
+ "dhat",
  "epoll",
  "event_monitor",
  "flume",
diff --git a/Cargo.toml b/Cargo.toml
index c1d8e4169..1de36fe48 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -63,7 +63,7 @@ wait-timeout = "0.2.0"
 [features]
 default = ["kvm", "io_uring"]
 dbus_api = ["zbus", "vmm/dbus_api"]
-dhat-heap = ["dhat"] # For heap profiling
+dhat-heap = ["dhat", "vmm/dhat-heap"] # For heap profiling
 guest_debug = ["vmm/guest_debug"]
 igvm = ["vmm/igvm", "mshv"]
 io_uring = ["vmm/io_uring"]
diff --git a/vmm/Cargo.toml b/vmm/Cargo.toml
index fc8520758..118c446c2 100644
--- a/vmm/Cargo.toml
+++ b/vmm/Cargo.toml
@@ -7,6 +7,7 @@ edition = "2021"
 [features]
 default = []
 dbus_api = ["blocking", "futures", "zbus"]
+dhat-heap = ["dhat"] # For heap profiling
 guest_debug = ["kvm", "gdbstub", "gdbstub_arch"]
 igvm = ["hex", "dep:igvm", "igvm_defs",  "mshv-bindings", "range_map_vec"]
 io_uring = ["block/io_uring"]
@@ -27,6 +28,7 @@ blocking = { version = "1.5.1", optional = true }
 cfg-if = "1.0.0"
 clap = "4.5.4"
 devices = { path = "../devices" }
+dhat = { version = "0.3.3", optional = true }
 epoll = "4.3.3"
 event_monitor = { path = "../event_monitor" }
 flume = "0.11.0"