Steering coupler mod

I received my steering coupler/U-joint. Steering box side is splined with a set screw. Shaft side is smooth. The end of the shaft still has the pin through it where the old coupler Plunges/engages. I was thinking of notching the new coupler to receive the pin on either side of the shaft. There is plenty of length to notch it. Thoughts?