Very simple fix......

Factory pot joint has provision for in and out movement, U-joint does not
However, the design of the steering shaft allows for movement. It is not one piece, but two pieces which allow enough movement. Rod