Modify lambda body as a block