Minor fix for brace highlighting