diff --git a/test/e2e/flask/btc/btc-account-overview.spec.ts b/test/e2e/flask/btc/btc-account-overview.spec.ts
index f32a48d9c4a8..418c9d736078 100644
--- a/test/e2e/flask/btc/btc-account-overview.spec.ts
+++ b/test/e2e/flask/btc/btc-account-overview.spec.ts
@@ -1,4 +1,3 @@
-import { strict as assert } from 'assert';
 import { Suite } from 'mocha';
 import { DEFAULT_BTC_BALANCE } from '../../constants';
 import { withBtcAccountSnap } from './common-btc';
@@ -46,17 +45,19 @@ describe('BTC Account - Overview', function (this: Suite) {
     await withBtcAccountSnap(
       { title: this.test?.fullTitle() },
       async (driver) => {
-        // Wait for the balance to load up
-        await driver.delay(2000);
-
-        const balanceElement = await driver.findElement(
-          '.coin-overview__balance',
-        );
-        const balanceText = await balanceElement.getText();
+        await driver.waitForSelector({
+          testId: 'account-value-and-suffix',
+          text: `${DEFAULT_BTC_BALANCE}`,
+        });
+        await driver.waitForSelector({
+          css: '.currency-display-component__suffix',
+          text: 'BTC',
+        });
 
-        const [balance, unit] = balanceText.split('\n');
-        assert(Number(balance) === DEFAULT_BTC_BALANCE);
-        assert(unit === 'BTC');
+        await driver.waitForSelector({
+          tag: 'p',
+          text: `${DEFAULT_BTC_BALANCE} BTC`,
+        });
       },
     );
   });