Fixed usage of class_attribute