Glad it helped. When I bought my heater they were including a 5 year parts and labor. I believe the new warranty is now 2 years labor and 7 years parts and 7 years compressor. If you bought your heater from a authorized dealer the parts should of been included.
Did they change your valve...