Fix save button