diff --git a/packages/kilo-vscode/src/KiloProvider.ts b/packages/kilo-vscode/src/KiloProvider.ts
index 7a43b8d1bb4..e9c6a09bd83 100644
--- a/packages/kilo-vscode/src/KiloProvider.ts
+++ b/packages/kilo-vscode/src/KiloProvider.ts
@@ -369,7 +369,7 @@ export class KiloProvider implements vscode.WebviewViewProvider, TelemetryProper
         case "requestFileSearch": {
           const client = this.httpClient
           if (client) {
-            const dir = this.getWorkspaceDirectory()
+            const dir = this.getWorkspaceDirectory(this.currentSession?.id)
             void client
               .findFiles(message.query, dir)
               .then((paths) => {