Fixed models.py bug