update wrapper class