Updated class name

Updated values to match the new class name