Little fix in EpidemicSimulator