diff --git a/tests/statistical/test_generate.py b/tests/statistical/test_generate.py
index c258acdf..aa1e1e40 100644
--- a/tests/statistical/test_generate.py
+++ b/tests/statistical/test_generate.py
@@ -24,7 +24,7 @@ class NextToken:
         probs: dict[int, List[float]] = {
             1: [0.2, 0.5, 0.3],
             2: [0.3, 0.4, 0.3],
-            3: [0, 0, 0]
+            3: [0, 0, 0],
         }
         p0: List[float] = [0.2, 0.8, 0.0]
         states: List[int] = [1, 2, 3]