Quick bug fix to models.py