diff --git a/chapter10/2-waitForLoad.py b/chapter10/2-waitForLoad.py
index 48c0ca5..d3231fa 100644
--- a/chapter10/2-waitForLoad.py
+++ b/chapter10/2-waitForLoad.py
@@ -1,3 +1,4 @@
+from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
@@ -10,4 +11,4 @@
     element = WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "loadedButton")))
 finally:
     print(driver.find_element_by_id("content").text)
-    driver.close()
\ No newline at end of file
+    driver.close()