tiny test refactoring