diff --git a/source/waits.rst b/source/waits.rst
index fc00b1e..1e0c50f 100644
--- a/source/waits.rst
+++ b/source/waits.rst
@@ -31,7 +31,7 @@ is one way this can be accomplished.
 
   from selenium import webdriver
   from selenium.webdriver.common.by import By
-  from selenium.webdriver.support.ui import WebDriverWait
+  from selenium.webdriver.support.wait import WebDriverWait
   from selenium.webdriver.support import expected_conditions as EC
 
   driver = webdriver.Firefox()