diff --git a/examples/pytorch/test_pytorch_examples.py b/examples/pytorch/test_pytorch_examples.py
index 1d4f8db9259087..9c86347f98c243 100644
--- a/examples/pytorch/test_pytorch_examples.py
+++ b/examples/pytorch/test_pytorch_examples.py
@@ -372,6 +372,7 @@ def test_run_translation(self):
             --predict_with_generate
             --source_lang en_XX
             --target_lang ro_RO
+            --max_source_length 512
         """.split()
 
         with patch.object(sys, "argv", testargs):