Improve robustness of relativstic Spheromak setup