Selenium test fix