Can you rework some footage from Jedi to make the visitation a dream?
If this were in writing I'd suggest him getting injured during the Death Star escape. You'd cut the TIE fighter battle and rush him to the infirmary on Hoth. While recovering, he has a dream of Obi-Wan telling him to complete his training.