Update models.py

Fixed wrong import in models.py