It's the old false ending, real ending trick. You can't have too much time between them, and you have to juggle the Lope being impregnated and the David/ Walter switcheroo. I'd have to watch it again to properly judge, but it didn't feel 'wrong' the first time.