Fixing selenium tests