fix for utils.wrap