this is what id be checkingIt could be a worn gearbox output shaft, there’s no lubrication on the splined output and it can wear out, causing loss of drive.
The failure is often preceded by a clunk when taking up drive.
There’s a modified replacement available that gives an oil feed to this part, preventing future failure.