Infernalist Demond form - bugged, spells forcing out of demon form

I was using demon form and fireball just fine.

Then Is started the boss fight with the lava flow in the gold place.

Now everytime I use fireball or living bomb i exit demon form. really frustrating when those are my core spells.

All other spells dont make me exit demon form, for some reason those do...