diff --git a/apps/sim/app/api/chat/edit/[id]/route.test.ts b/apps/sim/app/api/chat/edit/[id]/route.test.ts
index feee2f6d17..ee105a8850 100644
--- a/apps/sim/app/api/chat/edit/[id]/route.test.ts
+++ b/apps/sim/app/api/chat/edit/[id]/route.test.ts
@@ -71,7 +71,7 @@ describe('Chat Edit API Route', () => {
     }))
 
     vi.doMock('@/lib/urls/utils', () => ({
-      getBaseDomain: vi.fn().mockReturnValue('localhost:3000'),
+      getEmailDomain: vi.fn().mockReturnValue('localhost:3000'),
     }))
 
     vi.doMock('@/lib/environment', () => ({