Fix minor bug. Fix Selenium tests