Update models.py

fixed for 1.5