fixed failing test because of refactoring