diff --git a/tests/test_environments.py b/tests/test_environments.py
index 328fe07690..7a12bfd905 100644
--- a/tests/test_environments.py
+++ b/tests/test_environments.py
@@ -386,7 +386,7 @@ def test_cached_generate_batched(self):
             [True, True, True], past_key_values, past_attention_masks, past_input_ids
         )
 
-        input_texts2 = [" short interim", " a slightly longer interim", "another interim"]
+        input_texts2 = [" short interim", " a somewhat longer section in between", "something else entirely! So, "]
         model_inputs2 = [self.gpt2_tokenizer(txt, return_tensors="pt").input_ids.squeeze() for txt in input_texts2]
 
         outputs_cached, _, _, _, _ = env._generate_batched(