Improving $t \bar{t}$ reconstruction in the dilepton channel at future lepton colliders