fixed(tests): fix tests after refactor