Fix bugs in infect_recover