Nezha's augment mod, "Pyroclastic Flow" is pretty fun to use, but I find it just as awkward to use as well.
Every time I want to refresh the ability, it activates the augment, placing you in a brief and awkward animation lock that is completely unavoidable. Hence, there are moments where you may want to refresh the ability without utilizing the augment, but cannot.
I think a nice quality of life change would be making it so that the player would have to HOLD the ability's button before the augment is activated. Additionally, it can be made so that every time the ability refreshes, the damage that you have built up also refreshes. That way it cannot be abused and stacked indefinitely.
Just an idea. Would be nice.