fix models.py