diff --git a/src/e2e.test.ts b/src/e2e.test.ts
index 14caeb6..c7cdf4c 100644
--- a/src/e2e.test.ts
+++ b/src/e2e.test.ts
@@ -21,7 +21,7 @@ async function goToTrinoSettings(page: Page) {
 
 async function setupDataSourceWithAccessToken(page: Page) {
     await page.getByTestId('data-testid Datasource HTTP settings url').fill('http://trino:8080');
-    await page.locator('div').filter({hasText: /^Impersonate logged in userAccess token$/}).getByLabel('Toggle switch').click();
+    await page.locator('#trino-settings-enable-impersonation').click({ force: true });
     await page.locator('div').filter({hasText: /^Access token$/}).locator('input[type="password"]').fill('aaa');
     await page.getByTestId('data-testid Data source settings page Save and Test button').click();
 }