Update models.py

revert mistake commit