small refactor
refactor