models.py fix

type error