Add expectation of correct counts