small refactoring while reviewing