Minor fix in saving