Wrap the function