diff --git a/packages/pi-claude-cli/src/__tests__/acp-driver.test.ts b/packages/pi-claude-cli/src/__tests__/acp-driver.test.ts
index a6de14a60a..2c9ac962cb 100644
--- a/packages/pi-claude-cli/src/__tests__/acp-driver.test.ts
+++ b/packages/pi-claude-cli/src/__tests__/acp-driver.test.ts
@@ -1,4 +1,4 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
 import { EventEmitter } from "node:events";
 import { PassThrough } from "node:stream";