diff --git a/archivenow/handlers/is_handler.py b/archivenow/handlers/is_handler.py
index 2fc34ef..a3b16f5 100644
--- a/archivenow/handlers/is_handler.py
+++ b/archivenow/handlers/is_handler.py
@@ -50,18 +50,14 @@ def push(self, uri_org, p_args=[], session=requests.Session()):
             # The page takes a while to archive, so keep checking if the loading page is still displayed.
             loading = True
             while loading:
-                try:
-                    loadingPage = driver.find_element_by_xpath("/html/body/div/img")
-
-                except:
+                
+                if not 'wip' in driver.current_url:
                     loading = False
 
             # After the loading screen is gone and the page is archived, the current URL
             # will be the URL to the archived page.
             print(driver.current_url)
 
-            msg = driver.current_url
-
             driver.quit()
 
         except: