minor fix to models.py