diff --git a/tests/e2e/test_staging_full_saas.sh b/tests/e2e/test_staging_full_saas.sh
index b7d8ea1b7..8aa976e20 100755
--- a/tests/e2e/test_staging_full_saas.sh
+++ b/tests/e2e/test_staging_full_saas.sh
@@ -1011,7 +1011,7 @@ print(json.dumps({
             'messageId': f'e2e-{uuid.uuid4().hex[:8]}',
             'parts': [{'kind': 'text', 'text': 'Reply with exactly: ok'}],
         },
-        'configuration': {'max_tokens': 4}
+        'configuration': {'max_tokens': 32}
     }
 }))
 ")