Small style refactor