Very simple fix......

However, the design of the steering shaft allows for movement. It is not one piece, but two pieces which allow enough movement. Rod

Except the guys that engineered the system still used a steering joint that allowed for slip even with the 2 piece steering shaft. Because despite being 2 pieces that are supposed to collapse in case of an accident, it's not designed to take up that slip.

A simple fix isn't always the best.

This is what I did with mine. Not saying it's the best way to do it either, but it does account for the same movement that the factory joint can handle and a simple u-joint can't.
Steering shaft upgrade