update test wrap fn