Minor tidy test tidy up