Fixed broken imports in models.py

Lol Max