diff --git a/altair_saver/savers/_selenium.py b/altair_saver/savers/_selenium.py
index 0ac3208..50b3ce8 100644
--- a/altair_saver/savers/_selenium.py
+++ b/altair_saver/savers/_selenium.py
@@ -264,7 +264,10 @@ def _extract(self, fmt: str) -> MimebundleContent:
         driver.get("about:blank")
         driver.get(url)
         try:
-            driver.find_element_by_id("vis")
+            if hasattr(driver, 'find_element_by_id'):
+                driver.find_element_by_id("vis")
+            else:
+                driver.find_element(value="vis")
         except NoSuchElementException:
             raise RuntimeError(f"Could not load {url}")
         if not self._offline: