Small code readability refactoring