diff --git a/test/e2e/beta/metamask-beta-responsive-ui.spec.js b/test/e2e/beta/metamask-beta-responsive-ui.spec.js
index 98f0e02d5..099ed944e 100644
--- a/test/e2e/beta/metamask-beta-responsive-ui.spec.js
+++ b/test/e2e/beta/metamask-beta-responsive-ui.spec.js
@@ -278,6 +278,7 @@ describe('MetaMask', function () {
 
       const inputValue = await inputAmount.getAttribute('value')
       assert.equal(inputValue, '1')
+      await delay(regularDelayMs)
     })
 
     it('opens and closes the gas modal', async function () {
@@ -288,7 +289,7 @@ describe('MetaMask', function () {
 
       const gasModal = await driver.findElement(By.css('span .modal'))
 
-      const save = await findElement(driver, By.xpath(`//button[contains(text(), 'Save')]`))
+      const save = await findElement(driver, By.css('.page-container__header-close-text'))
       await save.click()
       await driver.wait(until.stalenessOf(gasModal), 10000)
       await delay(regularDelayMs)