Minor fix

Minor fix again