Any pump drive shaft that has a sharp cut where the round shaft transitions to the hex (see arrow pic below) is likely to snap under a heavy load, choose one that has a gradual transition.