Fixed concurrent modification issue with infection