Fixing some issues in the model