Correcting electrolyte imbalance