Houston, we have a problem. That is one wierd mod, from design and use. From here we can see that that spring is more of a hot spring than a switch return spring, mainly there to keep the battery in contact with the positive post. The pics won't copy well so here is the whole page:
LavaFire Extension
Your pics show just the one spring, and I thought it came out of the inside of the switch housing, rather than sit on the top of switch housing, especially the plastic part. I watched a whole 6:30 of the promo video, waiting to see the firing of the switch which the hot model never did...at least you get to hear the smoothness of those threads several times.
This part of the description bothers me:
The button can be screwed in and unscrewed out. After unscrew out, you can slip the button up and down, slip up to touch battery and get electricity, then you can inhale and vaporize; Slip down to stop vaporizing for seconds. When you need to stop vaporizing for long time, please remember to screw button in, then this button can not be slipped up and down, you don’t get fired any more, it is safe to put your LavaFire at anywhere.
I'm all for not getting fired anymore, but 'slip'?!?!? Not 'press' indicating some resistance for your efforts? I suppose the weight of the button itself (hailed as a sign of the high quality of materials used several times, which by the way is plated copper, yet a magnet lightly attracts to the alloy chosen) will successfully return the switch to the non-firing position, each and every time, arcing and corrosion be damned.
Now I understand much more precisely why you dislike this spring. Bzmoto, I assumed the action was too stiff, as do many others with switch springs. But yours just isn't doing anything at all switch related, so what is to like? If it were me, I would move this spring to the other end of the battery tube (with the batteries positive facing down) so it's compression is away from other moving parts. Unfortunately the nub on the other end probably won't hold it in place, this we can call a design feature. You need the plastic spacer for the switch anyway so just leave it where it is.
Back to the button. You've got it apart in the pics, but it is in the background and hard to make out with respect to details. I hope it is something threaded like a chicago screw, and the part it threads into has something to keep it from turning so that you consistently can open and close the safety without it just spinning. The post on the button is where the magnets go, and the post keeps them from flipping over. I hope the post fits the 1/8" hole, otherwise you will need some different mags. It may take some adjusting to get it right, maybe some plastic washers to get the distance right (cheaper than mags) and possibly shortening the spring. Sure you will see the mags in the switch now, but you can't see gravity.
I hope this helps, please let me know if I'm completely off about something.