From 25c1b9ab8fceaf963a89b5685a74085163eb74e9 Mon Sep 17 00:00:00 2001
From: jacob314 <jacob314@gmail.com>
Date: Sat, 7 Mar 2026 15:19:29 -0800
Subject: [PATCH] Fix ConfigInitDisplay flake

---
 .../cli/src/ui/components/ConfigInitDisplay.test.tsx     | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/packages/cli/src/ui/components/ConfigInitDisplay.test.tsx b/packages/cli/src/ui/components/ConfigInitDisplay.test.tsx
index 36ecbcbe5f..9802fd2a30 100644
--- a/packages/cli/src/ui/components/ConfigInitDisplay.test.tsx
+++ b/packages/cli/src/ui/components/ConfigInitDisplay.test.tsx
@@ -84,8 +84,9 @@ describe('ConfigInitDisplay', () => {
 
     // Wait for the UI to update
     await waitFor(() => {
-      expect(lastFrame()).toMatchSnapshot();
+      expect(lastFrame()).toContain('server2');
     });
+    expect(lastFrame()).toMatchSnapshot();
   });
 
   it('truncates list of waiting servers if too many', async () => {
@@ -120,8 +121,9 @@ describe('ConfigInitDisplay', () => {
     });
 
     await waitFor(() => {
-      expect(lastFrame()).toMatchSnapshot();
+      expect(lastFrame()).toContain('s1');
     });
+    expect(lastFrame()).toMatchSnapshot();
   });
 
   it('handles empty clients map', async () => {
@@ -147,7 +149,8 @@ describe('ConfigInitDisplay', () => {
     }
 
     await waitFor(() => {
-      expect(lastFrame()).toMatchSnapshot();
+      expect(lastFrame()).toContain('Initializing...');
     });
+    expect(lastFrame()).toMatchSnapshot();
   });
 });