Fix infection operator bug