diff --git a/ui-tests/tests/bigquery_dataset_explorer.spec.ts b/ui-tests/tests/bigquery_dataset_explorer.spec.ts
index c6cef154..1b4b839f 100644
--- a/ui-tests/tests/bigquery_dataset_explorer.spec.ts
+++ b/ui-tests/tests/bigquery_dataset_explorer.spec.ts
@@ -39,9 +39,9 @@ test('bigquery-dataset-explorer', async ({ page, request }) => {
 
     await page.locator('div[role="treeitem"].caret-icon.down').nth(1).click();
     await page.locator('div[role="treeitem"][aria-level="1"]').first().click();
-    await page.locator('div[role="treeitem"].caret-icon.down').nth(1).click();
+    await page.locator('div[role="treeitem"][aria-level="1"].caret-icon.down').first().click();
 
-    await page.locator('div[role="treeitem"][aria-level="2"]').first().click();
+    await page.locator('div[role="treeitem"][aria-level="2"]').third().click();
     await page.getByText('Schema', { exact: true }).click();
     await page.getByText('Preview', { exact: true }).click();
   } else {