For all the datasets used by stage-2 tuning, the context are provided as continuous paragraphs or documents that contain the answer. In contrast, the model needs to handle the top-\(k\) retrieved chunks at inference for long documents. To reduce such train/test mismatch, we investigate whether replacing some continuous paragraphs with retrieved top-k chunks will enhance the model's robustness.