fixing test count expectation