Minor fix to code style