During reassembly:

Insert the new head assembly up into the bearing as far as you can by hand.

Use a dead blow hammer to knock the head assembly up into the machine until there is no play in the shaft spacer.

Ensure there is no play in the shaft spacer before tightening the set screws onto the new head assembly.