fix class references