Fix the model declaration