fix: failed test after refactor