diff --git a/waternsw_grabber/waternsw_grabber.py b/waternsw_grabber/waternsw_grabber.py
index 07e7e24..27524e4 100644
--- a/waternsw_grabber/waternsw_grabber.py
+++ b/waternsw_grabber/waternsw_grabber.py
@@ -111,7 +111,8 @@ def get_telemetered_bore(driver, bore_id, start_date, end_date):
 
     # Close popup
     wait_for_element(driver, By.XPATH,
-                     "//div[contains(@class, 'lity-container')]")
+                     "//div[contains(@class, 'lity-container')]",
+                     timeout=30)
     webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()