Selenium test improvement