Update models.py

bad fix but don't know another way