Switch to eth and back before asserting on token balance (e2e beta token tests).

1 files changed, 2 insertions, 1 deletions

diff --git a/test/e2e/beta/metamask-beta-ui.spec.js b/test/e2e/beta/metamask-beta-ui.spec.js
index a8b273439..12a194e3a 100644
--- a/test/e2e/beta/metamask-beta-ui.spec.js
+++ b/test/e2e/beta/metamask-beta-ui.spec.js
@@ -778,7 +778,8 @@ describe('MetaMask', function () {
       const txStatuses = await findElements(driver, By.css('.tx-list-status'))
       await driver.wait(until.elementTextMatches(txStatuses[0], /Confirmed/))
 
-      // const walletBalance = await findElement(driver, By.css('.wallet-balance'))
+      const walletBalance = await findElement(driver, By.css('.wallet-balance'))
+      await walletBalance.click()
 
       const tokenListItems = await findElements(driver, By.css('.token-list-item'))
       await tokenListItems[0].click()