Very simple fix......
Factory pot joint has provision for in and out movement, U-joint does not
Print