I replaced the thermocouple. The flame is blue. It lights and stays lit, while the water is heating and then when the burner turns off, the pilot goes out. Does anybody know what could be causing this? Thanks.

Your themocouple is still bad need to replace it. Or the burnners are blowing it out how old is the water heater