diff --git a/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx b/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx
index 14b4b7e07a..2f58ad6e3c 100644
--- a/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx
+++ b/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx
@@ -31,6 +31,7 @@ jest.mock('~/hooks', () => ({
     },
     ref: { current: null },
   }),
+  useAuthContext: () => ({ token: 'mock-token' }),
 }));
 
 jest.mock('~/hooks/MCP', () => ({
@@ -89,6 +90,8 @@ jest.mock('lucide-react', () => ({
   ChevronDown: () => <span>{'ChevronDown'}</span>,
   ChevronUp: () => <span>{'ChevronUp'}</span>,
   TriangleAlert: () => <span>{'TriangleAlert'}</span>,
+  CheckCircle: () => <span>{'CheckCircle'}</span>,
+  XCircle: () => <span>{'XCircle'}</span>,
 }));
 
 jest.mock('~/utils', () => ({