Minor fix for style