Fixed the infection mechanism again.