Update class refs