The staked on nut will only be "loose" if the flange to output spline is worn.

In what direction are you seeing free play? Up / Down , rotation when looking at the flange to nut, rotation flange to trans housing when in park, ( this last one is normal )