Fix the bug of diff mode

The diff mode should not include the context.