Ensure that ml grading test looks for correct return value