Fix bug for late submission lab 3.