diff --git a/apps/sim/app/o/[organizationId]/settings/components/integrations/search-source-setup.test.tsx b/apps/sim/app/o/[organizationId]/settings/components/integrations/search-source-setup.test.tsx
index f9ad3c4022f..605ef2986ec 100644
--- a/apps/sim/app/o/[organizationId]/settings/components/integrations/search-source-setup.test.tsx
+++ b/apps/sim/app/o/[organizationId]/settings/components/integrations/search-source-setup.test.tsx
@@ -604,17 +604,21 @@ describe('organization setup entry points', () => {
         connector({ id: 'new-gmail', connectorType: 'gmail', accessMode: 'admin' })
       )
     )
-    expect(mocks.updateSearchIntegration).toHaveBeenCalledWith(
-      {
-        organizationId: 'org-1',
-        connectorType: 'gmail',
-        approved: true,
-        policy: expect.objectContaining({
-          accessMode: 'service_account',
-          sourceId: 'new-gmail',
-        }),
-      },
-      expect.any(Object)
+    await vi.waitFor(
+      () =>
+        expect(mocks.updateSearchIntegration).toHaveBeenCalledWith(
+          {
+            organizationId: 'org-1',
+            connectorType: 'gmail',
+            approved: true,
+            policy: expect.objectContaining({
+              accessMode: 'service_account',
+              sourceId: 'new-gmail',
+            }),
+          },
+          expect.any(Object)
+        ),
+      { interval: 1 }
     )
     expect(mocks.push).not.toHaveBeenCalled()
     await act(async () => mocks.updateSearchIntegration.mock.calls[0][1].onSuccess())