diff --git a/benchmark/packages/ipc/scripts/client.ts b/benchmark/packages/ipc/scripts/client.ts
index da74e40ae7..f00829a113 100644
--- a/benchmark/packages/ipc/scripts/client.ts
+++ b/benchmark/packages/ipc/scripts/client.ts
@@ -1,6 +1,7 @@
 import { TaskCommandName } from "@benchmark/types"
 
-import { IpcMessageType, IpcClient, IpcOrigin } from "../src/index.js"
+import { IpcMessageType, IpcOrigin } from "@benchmark/types"
+import { IpcClient } from "../src/index.js"
 
 async function main(socketPath: string, prompt: string) {
 	try {
diff --git a/src/extension.ts b/src/extension.ts
index 8684a6443b..5142f1ef42 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -115,8 +115,12 @@ export async function activate(context: vscode.ExtensionContext) {
 	registerCodeActions(context)
 	registerTerminalActions(context)
 
+	if (process.env.ROO_CODE_IPC_SOCKET_PATH) {
+		outputChannel.appendLine(`ROO_CODE_IPC_SOCKET_PATH: ${process.env.ROO_CODE_IPC_SOCKET_PATH}`)
+	}
+
 	// Implements the `RooCodeAPI` interface.
-	return new API(outputChannel, provider, "/tmp/roo-code-ipc.sock")
+	return new API(outputChannel, provider, process.env.ROO_CODE_IPC_SOCKET_PATH)
 }
 
 // This method is called when your extension is deactivated