chore: minor refactor