I understand that Joe and if you staple the NM down the angled beam of the ceiling then to the 2 x 4 to the closet point of the water heater you could meet that requirement you could also run a clip into the water heater with a tek screw close to the connector in order to have support within 12"
When was the last time you actually inspected a job as the AHJ?
