Moved to use assertHitCount instead of just checking the total count