diff --git a/tests/write-file.test.ts b/tests/write-file.test.ts
index 23cfe6f0..7d4a76d7 100644
--- a/tests/write-file.test.ts
+++ b/tests/write-file.test.ts
@@ -2,13 +2,19 @@ import { WriteFile } from "../src/tools/write-file";
 import { mkdirSync, writeFileSync, readFileSync, rmSync, existsSync } from "fs";
 import { join } from "path";
 import { tmpdir } from "os";
+import { Context } from "../src/types";
 
 describe("WriteFile", () => {
   let writeFile: WriteFile;
   let testDir: string;
+  let context: Context;
 
   beforeEach(() => {
-    writeFile = new WriteFile();
+    context = {
+      logger: console,
+    } as unknown as Context;
+
+    writeFile = new WriteFile(context);
     testDir = join(tmpdir(), `write-file-test-${Date.now()}`);
     mkdirSync(testDir, { recursive: true });
   });