most remote hwh switches are 12V running to a relay that switches the 120V to the element. If it really is 120V to the switch it should be #12 romex wire and a UL listed wall switch or toggle rated for 120V and at least 15A.
I haven't checked my suburban but i think the switch on the WH is also 12V switching a relay for the 120V.
BTW a friend had a brand new motor home where they had wired the switch to 120V rather than 12V and on walk through water heater was full, switched on the electric element and pretty quick smoke came rolling out. NOT a good thing.
here is a link to a typical suburban wiring diagram. Note NONE of the switches are 120V, they are ALL 12V.
https://answers.rvupgradestore.com/3519769/DO-YOU-HAVE-A-WIRING-DIAGRAM