diff --git a/ui/src/ui/app-gateway.sessions.node.test.ts b/ui/src/ui/app-gateway.sessions.node.test.ts
index 2ffc0959125a..385c6dfe59b0 100644
--- a/ui/src/ui/app-gateway.sessions.node.test.ts
+++ b/ui/src/ui/app-gateway.sessions.node.test.ts
@@ -8,6 +8,7 @@ const clearPendingQueueItemsForRunMock = vi.fn();
 const flushChatQueueForEventMock = vi.fn();
 const handleChatEventMock = vi.fn(() => "idle");
 const handleSessionOperationEventMock = vi.fn();
+const recordFirstAssistantChatTimingMock = vi.fn();
 
 vi.mock("./app-chat.ts", () => ({
   CHAT_SESSIONS_ACTIVE_MINUTES: 10,
@@ -39,6 +40,7 @@ vi.mock("./app-chat.ts", () => ({
   },
   clearPendingQueueItemsForRun: clearPendingQueueItemsForRunMock,
   flushChatQueueForEvent: flushChatQueueForEventMock,
+  recordFirstAssistantChatTiming: recordFirstAssistantChatTimingMock,
   refreshChatAvatar: vi.fn(),
 }));
 vi.mock("./app-settings.ts", () => ({