Feature should track counts in both test and train sets.