Minor style refactor