Fixed wrong import in models.py

Fixed wrong import in models.py