Small code style refactoring