Fix get_grades regression