Update models.py

Fix jsonfield compat