fix(tests): according to latest refactoring