Yeah, I wish Ben had survived the movie, but if there’s ever actually good canon material set after TROS, then I think the only real option which would maintain continuity would be to stick his ghost in and cut the kiss/adjust the music to sell his death as more of a premeditated heroic sacrifice. As-is, it felt to me like it was deliberately edited to be comically sudden and unexpected, like Meet the Spy or something.
There may be a way to make it a symmetry with Anakin’s force birth and have Rey be pregnant with Kylo’s kid through their dyad. He does touch her abdomen briefly before dying.