diff --git a/ui-tests/test/utils.ts b/ui-tests/test/utils.ts
index 1e71ac4ea..c307b2c97 100644
--- a/ui-tests/test/utils.ts
+++ b/ui-tests/test/utils.ts
@@ -6,9 +6,7 @@ import { IJupyterLabPageFixture } from '@jupyterlab/galata';
 export async function runAndAdvance(
   page: IJupyterLabPageFixture
 ): Promise<void> {
-  await page.click(
-    "//button[normalize-space(@title)='Run the selected cells and advance']"
-  );
+  await page.click(".jp-Toolbar-item [data-icon='ui-components:run']");
 }
 
 /**