You should be okay just replacing the UJoint. Just pay attention to the orientation of the flanges...mark them and put them in the same position.