Minor fix'