We had the same loss of drive on ours (same model) the spring does fit, but the tension only comes into play when the orange drive handle is pulled towards you, otherwise the mower would be in drive mode all the time. The problem appears to happen if you are trying to mower really thick grass which clogs up the spring and drive belt resulting in the spring being forced off.
Bob
June 2012