diff --git a/packages/tools/tests/img-gen.test.ts b/packages/tools/tests/img-gen.test.ts
index 0a74743b30367f9be7ec5bd2fc7333b2fb903b83..2f96cb44059e856646f957db10a23cbfd95aea4f 100644
--- a/packages/tools/tests/img-gen.test.ts
+++ b/packages/tools/tests/img-gen.test.ts
@@ -15,7 +15,7 @@ vi.mock("got", () => ({
 describe("Image Generator Tool", () => {
   test("generates image from prompt", async () => {
     const imgTool = imageGenerator({
-      apiKey: "mock-stability-key",
+      apiKey: process.env.STABILITY_API_KEY!,
       outputDir: "output",
       fileServerURLPrefix: "http://localhost:3000",
     });