diff --git a/litellm/tests/test_helicone_integration.py b/litellm/tests/test_helicone_integration.py
index 82669d0920..0348560009 100644
--- a/litellm/tests/test_helicone_integration.py
+++ b/litellm/tests/test_helicone_integration.py
@@ -26,5 +26,6 @@
 
 # # cohere call
 # response = completion(
-#     model="command-nightly", messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}]
+#     model="command-nightly",
+#     messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}],
 # )
diff --git a/litellm/utils.py b/litellm/utils.py
index d80d7d68f8..410f9ad882 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -1787,7 +1787,7 @@ class Logging:
                     if callback == "helicone":
                         print_verbose("reaches helicone for logging!")
                         model = self.model
-                        messages = kwargs["input"]
+                        messages = self.model_call_details["input"]
                         heliconeLogger.log_success(
                             model=model,
                             messages=messages,