fix save button issue