diff --git a/reasoning_gym/cognition/needle_haystack.py b/reasoning_gym/cognition/needle_haystack.py
index 179d0657..cc98765a 100644
--- a/reasoning_gym/cognition/needle_haystack.py
+++ b/reasoning_gym/cognition/needle_haystack.py
@@ -137,7 +137,7 @@ class NeedleHaystackCurriculum(BaseCurriculum):
         self._define_attributes(
             RangeAttributeDefinition(
                 name="num_statements",
-                levels=[10, 100, 1_000, 10_000, 100_000, 1_000_000, 168_386_000],
+                levels=[10, 100, 500, 1_000, 5_000, 10_000, 50_000, 100_000],
                 description="Number of statements in the haystack",
                 lower_field_name="min_num_statements",
                 upper_field_name="max_num_statements",