Update models.py

Fixed error