Fix reference to class