adjust unit test to refactoring