Fix staff grading test