Fix save button enabling after saving