I don’t think the point was that the teleporting healed the lightsaber. It’s that Ben turning back to the light and taking up his grandfather’s saber healed the crystal.
IMO, if there’s an audio/visual cue when Ben shrugs at the Knights of Ren, and the saber stabilizes at that point, it would work fine.