fix: tests after refactoring
fix: tests after refactoring