Minor fix for save button when loading data