Fix reference to class name