Fixed missing reference of class name