Very simple fix......

Factory pot joint has provision for in and out movement, U-joint does not