fix model
fix the model