diff --git a/rtl-spec/components/commands-version-chooser.spec.tsx b/rtl-spec/components/commands-version-chooser.spec.tsx
index 064ec7a415..87a5be3270 100644
--- a/rtl-spec/components/commands-version-chooser.spec.tsx
+++ b/rtl-spec/components/commands-version-chooser.spec.tsx
@@ -1,6 +1,6 @@
 import * as React from 'react';
 
-import { render, screen, waitFor } from '@testing-library/react';
+import { render, screen } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
 
 import { VersionChooser } from '../../src/renderer/components/commands-version-chooser';
@@ -28,10 +28,6 @@ describe('VersionSelect component', () => {
 
     await userEvent.click(versionButton);
 
-    waitFor(() =>
-      expect(btnOpenVersionSelector.textContent).toContain(
-        mockVersion1.version,
-      ),
-    );
+    expect(appState.setVersion).toHaveBeenCalledWith(mockVersion1.version);
   });
 });