Death wobble is usually a caster issue, compounded by other worn parts. A steering stabilizer can sometimes work as a short-term bandaid.
If you are going to swap in a newer axle, you'll be doing some suspension mods regardless, so you can add some more caster at the same time.