Minor bug fix in counting how many times a node has been infected