diff --git a/tests/graph-persistence.mjs b/tests/graph-persistence.mjs
index 2585831..4f23d76 100644
--- a/tests/graph-persistence.mjs
+++ b/tests/graph-persistence.mjs
@@ -228,6 +228,7 @@ async function createGraphPersistenceHarness({
     formatRecallContextLine,
     cloneGraphForPersistence,
     cloneRuntimeDebugValue,
+    onMessageReceivedController,
     getGraphPersistenceMeta,
     getGraphPersistedRevision,
     getGraphShadowSnapshotStorageKey,