Fix wrong count tests