fixed the save button issue