Fixed User model usage in models.py