If there is room to do so, try putting a stud, not a bolt into the output shaft, and secure it with a thread lock liquid like Loctite. Then use a nut to secure the universal joint. That way there won't be as much stress on the thread in the shaft as it will be a "one time" installation and the nut will be the part taking the tightening stress,
Jerry