Fix bug of "Save" button