Fixed models.py due to travis. The last fix, I hope