Update the model and remove the mask keyword