Update models.py

Поправил