Address jacob's feedback

Author: abhipatel12

packages/cli/src/ui/AppContainer.test.tsx

@@ -1515,13 +1515,13 @@ describe('AppContainer State Management', () => {
         />,
       );
 
+      // Verify initial model
       await act(async () => {
-        await new Promise((resolve) => setTimeout(resolve, 0));
+        await vi.waitFor(() => {
+          expect(capturedUIState?.currentModel).toBe('initial-model');
+        });
       });
 
-      // Verify initial model
-      expect(capturedUIState.currentModel).toBe('initial-model');
-
       // Get the registered handler for ModelChanged
       const handler = mockCoreEvents.on.mock.calls.find(
         (call: unknown[]) => call[0] === CoreEvent.ModelChanged,