diff --git a/pysplit/traj.py b/pysplit/traj.py index 8eb70de..16c04ef 100644 --- a/pysplit/traj.py +++ b/pysplit/traj.py @@ -439,13 +439,14 @@ def moisture_uptake(self, precipitation, evaporation, interval=6, if self.uptake.loc[w, 'dq_initial'] < precipitation: # Adjust previous fractions - self.uptake.loc[is_below, 'dq'] = ( - self.uptake.loc[is_below, 'below'] * - self.uptake.loc[w, 'q']) - - self.uptake.loc[is_above, 'dq'] = ( - self.uptake.loc[is_above, 'above'] * - self.uptake.loc[w, 'q']) + if self.uptake.loc[w, 'Timestep'] != 0: + self.uptake.loc[is_below, 'dq'] = ( + self.uptake.loc[is_below, 'below'] * + self.uptake.loc[w, 'q']) + + self.uptake.loc[is_above, 'dq'] = ( + self.uptake.loc[is_above, 'above'] * + self.uptake.loc[w, 'q']) def load_clippedtraj_data(self, clipped_dir='default', fname_end='CLIPPED'):