diff --git a/turbo_alignment/metrics/distinctness.py b/turbo_alignment/metrics/distinctness.py
index 101a222..f5ee2fa 100755
--- a/turbo_alignment/metrics/distinctness.py
+++ b/turbo_alignment/metrics/distinctness.py
@@ -41,7 +41,7 @@ def compute(self, **kwargs) -> list[MetricResults]:
     @staticmethod
     def distinctness(answers: list[str], vocab_size: int, ngram: int) -> dict[str, float]:
         ngram_sets: list[set] = [set() for _ in range(ngram)]
-        total_ngrams = [0] * ngram
+        total_ngrams: list[int] = [0] * ngram
 
         for answer in answers:
             words = answer.split(' ')