diff --git a/packages/ui/src/components/Visualization/ContextToolbar/DSLSelector/DSLSelectorToggle/DSLSelectorToggle.test.tsx b/packages/ui/src/components/Visualization/ContextToolbar/DSLSelector/DSLSelectorToggle/DSLSelectorToggle.test.tsx
index aa4a648c6..52c65485f 100644
--- a/packages/ui/src/components/Visualization/ContextToolbar/DSLSelector/DSLSelectorToggle/DSLSelectorToggle.test.tsx
+++ b/packages/ui/src/components/Visualization/ContextToolbar/DSLSelector/DSLSelectorToggle/DSLSelectorToggle.test.tsx
@@ -99,8 +99,9 @@ describe('DSLSelectorToggle.tsx', () => {
     });
 
     const element = await wrapper.findByText('Pipe');
-    expect(element).toBeInTheDocument();
-
+    waitFor(() => {
+      expect(element).toBeInTheDocument();
+    });
     /** Close Select */
     act(() => {
       fireEvent.click(toggle);