Updated to correct the name of a class preventing compilation