I do a couple of things.
Mostly I just grab a stool and a cup of coffee and start turning the shaft a few degrees at a time with slight pressure applied. Just when you think this is a stupid idea, the shaft slides easily in place.
The other method I have used is file some approach ramps on the leading edge of the pump and flywheel splines. Seems to help a lot.
Good luck, sometimes it takes awhile.