Fix models.py for real