Fixes last failing auto-grader test