Minor fix to format