fix wrap_class