diff --git a/tests/global.d.ts b/tests/global.d.ts
index ccc490a..7552ffc 100644
--- a/tests/global.d.ts
+++ b/tests/global.d.ts
@@ -1,8 +1,8 @@
-import {ModelContext, ToolDefinition} from '../src/webmcp_types';
+import {ModelContext, WebMcpTool} from '../src/webmcp_types';
 
 declare global {
   interface Window {
-    __registeredTools?: ToolDefinition[];
+    __registeredTools?: WebMcpTool[];
   }
 
   interface Navigator {
diff --git a/tests/webmcp.spec.ts b/tests/webmcp.spec.ts
index 6647ae8..657eaff 100644
--- a/tests/webmcp.spec.ts
+++ b/tests/webmcp.spec.ts
@@ -1,5 +1,5 @@
 import {expect, test} from '@playwright/test';
-import {ToolDefinition} from '../src/webmcp_types';
+import {WebMcpTool} from '../src/webmcp_types';
 import {setupGedcomRoute} from './helpers';
 
 const EXPECTED_TOOL_NAMES = [
@@ -18,10 +18,10 @@ test.describe('WebMCP Integration', () => {
 
     // Add init script to expose modelContext mock BEFORE application boots.
     await page.addInitScript(() => {
-      const registeredTools: ToolDefinition[] = [];
+      const registeredTools: WebMcpTool[] = [];
       window.__registeredTools = registeredTools;
       window.navigator.modelContext = {
-        registerTool: (tool: ToolDefinition) => {
+        registerTool: (tool: WebMcpTool) => {
           registeredTools.push(tool);
         },
         unregisterTool: (name: string) => {