Update models.py

Fix for django 1.6