Missing an import here? from selenium.webdriver.support.ui import WebDriverWait?