diff --git a/apps/desktop/stories/settings/provider-settings.stories.tsx b/apps/desktop/stories/settings/provider-settings.stories.tsx
index c0c5ee61ec..8c2ac7789c 100644
--- a/apps/desktop/stories/settings/provider-settings.stories.tsx
+++ b/apps/desktop/stories/settings/provider-settings.stories.tsx
@@ -903,7 +903,7 @@ export const OAuthCreateAdoptsExactConnection: Story = {
       '[data-connection-id="connection-openai-codex-4"]',
     );
     await expect(createdRow).not.toBeNull();
-    await expect(within(createdRow!).getByRole('button')).toHaveFocus();
+    await waitFor(() => expect(within(createdRow!).getByRole('button')).toHaveFocus());
   },
 };