Fixed broken class references.