diff --git a/packages/client/src/client.spec.ts b/packages/client/src/client.spec.ts
index 69237b5..440f752 100644
--- a/packages/client/src/client.spec.ts
+++ b/packages/client/src/client.spec.ts
@@ -60,7 +60,7 @@ const collabBackend = new AutomergeCollaboration({
   async onDocumentLoad(pathname) {
     return defaultSlateJson
   },
-  onTrace(metaData, computationFn) {
+  onTrace(metaData, socket, computationFn) {
     operationTraces.push(metaData)
     computationFn()
   }