Minor fix

Minor fix