Fixed incorrect class name