Fixed the automated testing bug on models.py