class reference fix