diff --git a/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb b/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb
index 0917211063..5406de18b1 100644
--- a/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb
+++ b/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb
@@ -205,11 +205,11 @@ class VisualEditorPage
       loop do
         begin
           browser.execute_script script
-        rescue Selenium::WebDriver::Error::JavascriptError => e
+        rescue Selenium::WebDriver::Error::JavascriptError, Selenium::WebDriver::Error::UnknownError => e
         else
           e = nil
         end
-        break unless e.class == Selenium::WebDriver::Error::JavascriptError
+        break unless (e.class == Selenium::WebDriver::Error::JavascriptError) || (e.class == Selenium::WebDriver::Error::UnknownError)
       end
     end
     browser.execute_script script