BUG:

I didn't choose an element before choosing an attack (after doing so before, in the menu) and the element stays. (It should be reset after an attack, but not before the UI shows - I think you forgot to reset it while the player is in the menu, instead of only resetting after defeating an enemy.)