Forgot the wrapping code during the latest release