fix the model after validation