The odd thing about water heater internal wiring is that it has no markings on it, on the ones I look at. We might assume that it is 190'F or more insulation, but only the MFG seems to know, and maybe UL.

Which would mean that only THHN wire should terminate in the heater from the feed. But I see no code on that.