Fix a bug where the incorrect variable (total_nr_timesteps_per_baseline) was used.
total_nr_timesteps_per_baseline