Minor tweak to selenium tests