adding small test refactoring