Rewrite the callback and the return body