diff --git a/src/infra/heartbeat-wake.test.ts b/src/infra/heartbeat-wake.test.ts
index 7718b931601..66e3a99e2d6 100644
--- a/src/infra/heartbeat-wake.test.ts
+++ b/src/infra/heartbeat-wake.test.ts
@@ -7,6 +7,13 @@ import {
   setHeartbeatWakeHandler,
 } from "./heartbeat-wake.js";
 
+async function loadWakeModule() {
+  return {
+    requestHeartbeatNow,
+    setHeartbeatWakeHandler,
+  };
+}
+
 describe("heartbeat-wake", () => {
   beforeEach(() => {
     resetHeartbeatWakeStateForTests();