I think its fine how it is, its about tactics on how to transform safely, they are already immune to melle while transformin which is good i think, when they see a beam comin, they can choose to risk it and keep on transformin and hope you survive the attack or cancel the tarnsform and try to move somehwere else safer and try again.