Fixed the selenium tests