The goal in replacing it was to avoid spoiling the end of the film. Also, to avoid a similar thing to what you described. Leia sacrifices her life to bring back Ben, and if she knew that would result in his death as well due to it requiring her to be a Jedi again it muddies the waters a bit. It’s better for Ben’s sacrifice to be of his own agency rather than because of some vision his mom had 30 years ago.
Dom, here is the line I was referring to:
https://youtu.be/-LzqBD8--H4