Minor fix to saving