A stub shaft of a main rotor shaft of a generator has its flange or shoulder removed from its outer portion so in the event of a bending failure of the stub shaft the resulting axial load of the stub shaft is not transmitted to the main rotor shaft. Instead of a retention plug adjacent the distal end of the inside portion of the stub shaft, a knock out plug is lightly press-fit into the main rotor shaft. In the event of a failure, the outside portion of the stub shaft enters the main rotor shaft without applying an axial load onto the main rotor shaft, moves axially and displaces the knock-out plug without exiting the main rotor shaft.