diff --git a/e2e-tests/utils/walletPageHelper.ts b/e2e-tests/utils/walletPageHelper.ts
index 091899a895..b46c07b792 100644
--- a/e2e-tests/utils/walletPageHelper.ts
+++ b/e2e-tests/utils/walletPageHelper.ts
@@ -75,9 +75,10 @@ export default class WalletPageHelper {
       .last()
       .click({ trial: true })
 
-    // There's no `connection_button` in the current code from `main`. TODO:
-    // verify if we'll need this or not.
-    // await this.popup.locator(".connection_button").last().click({ trial: true })
+    await expect(
+      this.popup.getByText("Connect by website using:")
+    ).toBeVisible()
+    await this.popup.locator(".bulb").last().click({ trial: true })
 
     await expect(
       this.popup.getByTestId("top_menu_profile_button").last()