fixed error in models.py