diff --git a/libs/community/langchain_community/llms/mlflow.py b/libs/community/langchain_community/llms/mlflow.py
index 8c4c1118d9..b8f733d720 100644
--- a/libs/community/langchain_community/llms/mlflow.py
+++ b/libs/community/langchain_community/llms/mlflow.py
@@ -33,7 +33,7 @@ class Mlflow(LLM):
             completions = Mlflow(
                 target_uri="http://localhost:5000",
                 endpoint="test",
-                params={"temperature": 0.1}
+                temperature=0.1,
             )
     """