models.py bugfix