diff --git a/tests/conftest.py b/tests/conftest.py
index 66106d1bf7792..c61a8f8dd539a 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -756,7 +756,7 @@ class VllmRunner:
 
     def get_inputs(
         self,
-        prompts: Union[list[str], list[torch.Tensor], list[int]],
+        prompts: Union[list[str], list[torch.Tensor], list[list[int]]],
         images: Optional[PromptImageInput] = None,
         videos: Optional[PromptVideoInput] = None,
         audios: Optional[PromptAudioInput] = None,