fix the class reference