Slight change

Changed a class name