Yes, a 2-gallon expansion tank is generally sufficient for a 50-gallon water heater for typical usage.