Fix class reference