Refactor proof for stronger lemma.