diff --git a/superset-frontend/src/explore/components/controls/DndColumnSelectControl/DndColumnSelect.test.tsx b/superset-frontend/src/explore/components/controls/DndColumnSelectControl/DndColumnSelect.test.tsx
index bc837203dbc4b..13996091e105c 100644
--- a/superset-frontend/src/explore/components/controls/DndColumnSelectControl/DndColumnSelect.test.tsx
+++ b/superset-frontend/src/explore/components/controls/DndColumnSelectControl/DndColumnSelect.test.tsx
@@ -33,23 +33,23 @@ const defaultProps: DndColumnSelectProps = {
actions: { setControlValue: jest.fn() },
};
-test('renders with default props', () => {
+test('renders with default props', async () => {
render(, {
useDnd: true,
useRedux: true,
});
- expect(screen.getByText('Drop columns here')).toBeInTheDocument();
+ expect(await screen.findByText('Drop columns here')).toBeInTheDocument();
});
-test('renders with value', () => {
+test('renders with value', async () => {
render(, {
useDnd: true,
useRedux: true,
});
- expect(screen.getByText('Column A')).toBeInTheDocument();
+ expect(await screen.findByText('Column A')).toBeInTheDocument();
});
-test('renders adhoc column', () => {
+test('renders adhoc column', async () => {
render(
{
/>,
{ useDnd: true, useRedux: true },
);
- expect(screen.getByText('adhoc column')).toBeVisible();
+ expect(await screen.findByText('adhoc column')).toBeVisible();
expect(screen.getByLabelText('calculator')).toBeVisible();
});