diff --git a/lm/builder/adjust_counts.cc b/lm/builder/adjust_counts.cc index b3ae1192..e4ad668e 100644 --- a/lm/builder/adjust_counts.cc +++ b/lm/builder/adjust_counts.cc @@ -335,7 +335,7 @@ void AdjustCounts::Run(const util::stream::ChainPositions &positions) { } } - stats.Add(s - streams.begin(), lower_count, (*s)->Value().IsMarked()); + stats.Add(s - streams.begin(), (*s)->Value().UnmarkedCount(), (*s)->Value().IsMarked()); ++*s; } // Poison everyone! Except the N-grams which were already poisoned by the input.