Fix the model names