diff --git a/litellm/tests/test_langsmith.py b/litellm/tests/test_langsmith.py
index c4d0e84f53..9575c18da7 100644
--- a/litellm/tests/test_langsmith.py
+++ b/litellm/tests/test_langsmith.py
@@ -60,8 +60,9 @@ async def test_async_langsmith_logging():
         print("fields in logged_run_on_langsmith", logged_run_on_langsmith.keys())
 
         input_fields_on_langsmith = logged_run_on_langsmith.get("inputs")
-        _extra = logged_run_on_langsmith.get("extra", {}) or {}
-        extra_fields_on_langsmith = _extra.get("invocation_params")
+        extra_fields_on_langsmith = logged_run_on_langsmith.get("extra").get(
+            "invocation_params"
+        )
 
         print("\nLogged INPUT ON LANGSMITH", input_fields_on_langsmith)
 
@@ -164,8 +165,9 @@ async def test_async_langsmith_logging_with_streaming_and_metadata(sync_mode):
 
         input_fields_on_langsmith = logged_run_on_langsmith.get("inputs")
 
-        _extra = logged_run_on_langsmith.get("extra", {}) or {}
-        extra_fields_on_langsmith = _extra.get("invocation_params")
+        extra_fields_on_langsmith = logged_run_on_langsmith.get("extra").get(
+            "invocation_params"
+        )
 
         assert logged_run_on_langsmith.get("run_type") == "llm"
         print("\nLogged INPUT ON LANGSMITH", input_fields_on_langsmith)