Fixed some custom class references.