diff --git a/src/vscode/host.ts b/src/vscode/host.ts
index 0a41632..e8e2c48 100644
--- a/src/vscode/host.ts
+++ b/src/vscode/host.ts
@@ -17,7 +17,7 @@ import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
 import { FullConfig } from '../config.js';
 import { ClientFactory } from '../mcp/proxyBackend.js';
-import type { Implementation } from '@modelcontextprotocol/sdk/types.js';
+import { Server } from '../mcp/server.js';
 
 class VSCodeClientFactory implements ClientFactory {
   name = 'vscode';
@@ -25,13 +25,13 @@ class VSCodeClientFactory implements ClientFactory {
 
   constructor(private readonly _config: FullConfig) {}
 
-  async create(clientVersion: Implementation, options: any): Promise<Client> {
+  async create(server: Server, options: any): Promise<Client> {
     if (typeof options.connectionString !== 'string')
       throw new Error('Missing options.connectionString');
     if (typeof options.lib !== 'string')
       throw new Error('Missing options.library');
 
-    const client = new Client(clientVersion);
+    const client = new Client(server.getClientVersion()!);
     await client.connect(new StdioClientTransport({
       command: process.execPath,
       cwd: process.cwd(),