fixed issue with new webdriver