diff --git a/index.d.ts b/index.d.ts
index 551ed55..9ea8bcc 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -16,13 +16,11 @@
  */
 
 import type { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import type { Config } from './config';
-import type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import type { Config } from './config.js';
 import type { BrowserContext } from 'playwright';
 
 export type Connection = {
   server: Server;
-  connect(transport: Transport): Promise<void>;
   close(): Promise<void>;
 };
 
diff --git a/src/index.ts b/src/index.ts
index 3c865be..2f5f2f9 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -14,7 +14,8 @@
  * limitations under the License.
  */
 
-import { Connection, createConnection as createConnectionImpl } from './connection.js';
+import { createConnection as createConnectionImpl } from './connection.js';
+import type { Connection } from '../index.js';
 import { resolveConfig } from './config.js';
 import { contextFactory } from './browserContextFactory.js';