Fixed bug in submission.py