Prophet Headset Upgrades

I've got a Prophet 4 and I'm feeling some small axial play so I'm thinking it's time for a new headset. I'm still running the RS Tora fork, so what's a good upgraded headset? Cane Creek? FSA? Or stick with a Cannondale headset? I ride on rocky trails so I'm looking for something that will last and can handle the abuse.

Thanks for the suggestion! Since I posted this thread I ended up going with a FSA headset since I put a tapered fork on my Prophet and it was the easiest for me to mix and match the top and bottom cup.