diff --git a/environments/math_server_zero.py b/environments/math_server_zero.py
index d543b7e7..b067728e 100644
--- a/environments/math_server_zero.py
+++ b/environments/math_server_zero.py
@@ -259,7 +259,7 @@ class MathEnv(BaseEnv):
             completion = await managed.completion(
                 prompt=question,
                 n=1,
-                max_tokens=32765,
+                max_tokens=self.config.max_token_length,
                 temperature=0.0,
                 split="eval",
                 stop=stop_list,