diff --git a/apps/cli/src/api.ts b/apps/cli/src/api.ts
index 599581cf..3b3fc569 100644
--- a/apps/cli/src/api.ts
+++ b/apps/cli/src/api.ts
@@ -39,9 +39,7 @@ async function taskHandler(
 ): Promise<StepHandler> {
 
   const workspace = new AgentProtocolWorkspace(
-    path.join(
-      process.env.AGENT_WORKSPACE as string, id
-    )
+    path.join(workspaceDir, id)
   );
   const app = createApp({
     rootDir,