diff --git a/test/e2e/tests/chain-interactions.spec.js b/test/e2e/tests/chain-interactions.spec.js
index 4e474ef4e..3135ab168 100644
--- a/test/e2e/tests/chain-interactions.spec.js
+++ b/test/e2e/tests/chain-interactions.spec.js
@@ -55,10 +55,10 @@ describe('Chain Interactions', function () {
         await driver.switchToWindow(extension);
 
         // verify networks
-        const networkDisplay = await driver.findElement(
-          '[data-testid="network-display"] p',
-        );
-        assert.equal(await networkDisplay.getText(), 'Localhost 8545');
+        await driver.findElement({
+          css: '[data-testid="network-display"]',
+          text: 'Localhost 8545',
+        });
 
         await driver.clickElement('[data-testid="network-display"]');
         const ganacheChain = await driver.findElements({
@@ -103,10 +103,10 @@ describe('Chain Interactions', function () {
         await driver.switchToWindow(extension);
 
         // verify current network
-        const networkDisplay = await driver.findElement(
-          '[data-testid="network-display"] p',
-        );
-        assert.equal(await networkDisplay.getText(), `Localhost ${port}`);
+        await driver.findElement({
+          css: '[data-testid="network-display"]',
+          text: `Localhost ${port}`,
+        });
       },
     );
   });
diff --git a/test/e2e/tests/onboarding.spec.js b/test/e2e/tests/onboarding.spec.js
index f1ea9b262..d0f494fd7 100644
--- a/test/e2e/tests/onboarding.spec.js
+++ b/test/e2e/tests/onboarding.spec.js
@@ -301,10 +301,10 @@ describe('MetaMask onboarding', function () {
         assert.equal(networkNotification, true);
 
         // Check localhost 8546 is selected and its balance value is correct
-        const networkDisplay = await driver.findElement(
-          '[data-testid="network-display"] p',
-        );
-        assert.equal(await networkDisplay.getText(), networkName);
+        await driver.findElement({
+          css: '[data-testid="network-display"]',
+          text: networkName,
+        });
 
         await assertAccountBalanceForDOM(driver, secondaryGanacheServer);
       },