WIP: fix grading: mini fix