Do you not need to check `originalRegistrationEntity` anymore? You are still doing so in `recordCompletedTimestampIfThisUpdateCompletesRegistration`.
As a side note, if you intend to continue to use `originalRegistrationEntity`, I would pass it as a parameter instead of setting private class data member.