diff --git a/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py b/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py
index d156f9e6fbd67..111ef82bff469 100644
--- a/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py
+++ b/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py
@@ -383,7 +383,7 @@ def __init__(
         self,
         llm: TritonTensorRTLLM,
         request_id: str,
-        force_batch: bool,
+        signal_stop: bool,
         stop_words: Sequence[str],
     ) -> None:
         """Instantiate the generator class."""