Update class references