Fixed compatibility with the new Infected API.