diff --git a/packages/cli/e2e-test/helpers.js b/packages/cli/e2e-test/helpers.js
index 073248789e..4d3f4f4661 100644
--- a/packages/cli/e2e-test/helpers.js
+++ b/packages/cli/e2e-test/helpers.js
@@ -130,15 +130,14 @@ async function waitForPageWithText(
       );
       break;
     } catch (error) {
-      if (error instanceof browser.assert.AssertionError) {
-        findTextAttempts++;
-        if (findTextAttempts <= maxFindTextAttempts) {
-          await browser.visit(path);
-          await new Promise(resolve => setTimeout(resolve, intervalMs));
-          continue
-        }
+      findTextAttempts++;
+      if (findTextAttempts <= maxFindTextAttempts) {
+        await browser.visit(path);
+        await new Promise(resolve => setTimeout(resolve, intervalMs));
+        continue
+      } else {
+        throw error;
       }
-      throw error;
     }
   }
 }