Fix atmosphere model tests.
More fixes needed to support new mass balance units.