Updated / Fixed incorrect class name