Correction to RandomInfectedSampler

Correction to RandomInfectedSampler, but not compiled yet, or tested