Fixes bug with the save button