I hate to 'stir the pot,' but,
But.....you will.
Well flexible cords are not typically allowed for taking the place of premise wiring systems.
Also connectors are a nice added feature that cord is missing but the NM in your early picture has one.
As others have said NM directly into a water heater is a fairly common install and I have yet to see one fail or be damaged.
