diff --git a/playwriter/package.json b/playwriter/package.json
index 2a6469f..3f8503d 100644
--- a/playwriter/package.json
+++ b/playwriter/package.json
@@ -29,6 +29,7 @@
     "@mizchi/selector-generator": "1.50.0-next",
     "@types/node": "^24.10.1",
     "@vitest/ui": "^4.0.8",
+    "mcp-extension": "workspace:*",
     "vite-node": "^5.0.0",
     "vitest": "^4.0.8"
   },
diff --git a/playwriter/src/mcp.test.ts b/playwriter/src/mcp.test.ts
index c5307d1..5d65522 100644
--- a/playwriter/src/mcp.test.ts
+++ b/playwriter/src/mcp.test.ts
@@ -7,6 +7,7 @@ import path from 'node:path'
 import fs from 'node:fs'
 import os from 'node:os'
 import { getCdpUrl } from './utils.js'
+import type { ExtensionState } from 'mcp-extension/src/types.js'
 
 import { spawn } from 'node:child_process'
 
@@ -51,8 +52,8 @@ async function killProcessOnPort(port: number): Promise<void> {
 }
 
 declare global {
-    var toggleExtensionForActiveTab: () => Promise<{ isConnected: boolean; state: any }>;
-    var getExtensionState: () => { connectedTabs: Map<number, { targetId: string; state: string }>; connectionState: string };
+    var toggleExtensionForActiveTab: () => Promise<{ isConnected: boolean; state: ExtensionState }>;
+    var getExtensionState: () => ExtensionState;
     var disconnectEverything: () => Promise<void>;
     var chrome: any;
 }