Fixed models.py due to travis