Fixed logic in set infected