fixed logic for total_infection