Fixing a bug in infectionIs