Update InfectionBehavior.scala