fixed test refactoring

:-)