Fix: convert class name to class