small refactor
small refactoring