Minor fix for saving