To identify and fix a faulty thermostat, you must first test it electrically to make sure that it is the source of the problem. Once you have identified the issue, you can then replace the thermostat with a new one that is compatible with your AO Smith water heater.