little test refactor