This is more of a suggestion but I feel like the elemental forms should not be able to be insta toggled off whenever or else this defeats the purpose of having drawbacks to the elemental forms. The elemental forms should leave you in one of the forms for a while so you avoid going to warm biomes as a water elemental and avoid cold biomes as a fire elemental.