Minor fix in ending