Small changes to selenium test