If this is a new installation, wouldn't the romex be Orange ?
Personally,I wouldn't install this without a Disconnect. They are cheap,& make a nice transition to a short piece of flex for the Heater.
That last 3-4 ft of NM just doesn't look neat, & secure when it goes straight to the water heater.