A 2-gallon expansion tank is the adequate size for water heaters that are 40- to 50-gallons.