Fix the model name