Fix attribute_set class