Small test refactoring.