diff --git a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
index c30aa89f113..e22e99b0aeb 100644
--- a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
+++ b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
@@ -149,7 +149,7 @@ def test_inline_pdf_tokens_costed_when_text_tokens_less_than_prompt_tokens():
         usage= usage,
         model = "gemini-2.0-flash-001",
     )
-    cost = response_cost_calculator(
+from litellm.cost_calculator import response_cost_calculator
         response_object= response,
         model = "gemini-2.0-flash-001",
         custom_llm_provider = "vertex_ai",