Update models.py

little fix