diff --git a/tests/integration/train_tests.py b/tests/integration/train_tests.py
index 31889a51b2..e984b58667 100644
--- a/tests/integration/train_tests.py
+++ b/tests/integration/train_tests.py
@@ -689,6 +689,7 @@ def test_fractional_eval_batch_size(self):
         "steps=5",
         "enable_checkpointing=False",
         "enable_goodput_recording=False",
+        "max_target_length=4096",
         "dataset_type=synthetic",
         "remat_policy=minimal",
         "per_device_batch_size=1",