diff --git a/packages/@n8n/instance-ai/src/tools/orchestration/tracing-utils.ts b/packages/@n8n/instance-ai/src/tools/orchestration/tracing-utils.ts
index b79ca284e21..0b7ba2b6ab1 100644
--- a/packages/@n8n/instance-ai/src/tools/orchestration/tracing-utils.ts
+++ b/packages/@n8n/instance-ai/src/tools/orchestration/tracing-utils.ts
@@ -1,5 +1,3 @@
-import type { ToolsInput } from '@mastra/core/agent';
-
 import {
 	createDetachedSubAgentTraceContext,
 	mergeCurrentTraceMetadata,
@@ -11,6 +9,8 @@ import type {
 	OrchestrationContext,
 } from '../../types';
 
+type ToolRegistry = OrchestrationContext['domainTools'];
+
 interface StartSubAgentTraceOptions {
 	agentId: string;
 	role: string;
@@ -100,9 +100,9 @@ export async function createDetachedSubAgentTracing(
 
 export function traceSubAgentTools(
 	context: OrchestrationContext,
-	tools: ToolsInput,
+	tools: ToolRegistry,
 	role: string,
-): ToolsInput {
+): ToolRegistry {
 	return (
 		context.tracing?.wrapTools(tools, {
 			agentRole: role,