Update incorrect class name