Hey Dradam,
I'm going to do the same to my heater and when I did the google research and found your post.
Not sure if that's the root cause to your issue. But based on this manual, it's recommended to have a 120V normal close solenoid valve installed to avoid colder than desired DHW...