Webdriver element fix

Hey oleg!