Small update to manage.py