Close popup window for e2e beta token factory send test.

1 files changed, 3 insertions, 2 deletions

diff --git a/test/e2e/beta/metamask-beta-ui.spec.js b/test/e2e/beta/metamask-beta-ui.spec.js
index 36295adac..9233b9c60 100644
--- a/test/e2e/beta/metamask-beta-ui.spec.js
+++ b/test/e2e/beta/metamask-beta-ui.spec.js
@@ -720,8 +720,10 @@ describe('MetaMask', function () {
       await transferAmountButton.click()
       await delay(regularDelayMs)
 
+      const [,, popup] = await driver.getAllWindowHandles()
+      await driver.switchTo().window(popup)
+      await driver.close()
       await driver.switchTo().window(extension)
-      await loadExtension(driver, extensionId)
       await delay(regularDelayMs)
 
       // Set the gas limit
@@ -748,7 +750,6 @@ describe('MetaMask', function () {
       if (process.env.SELENIUM_BROWSER === 'firefox') {
         await gasLimitInput.sendKeys(Key.BACK_SPACE)
       }
-      await delay(tinyDelayMs)
 
       const save = await findElement(driver, By.css('.send-v2__customize-gas__save'))
       await save.click()