diff --git a/litellm/proxy/db/db_spend_update_writer.py b/litellm/proxy/db/db_spend_update_writer.py
index d5f2ef5228..af1d3294c5 100644
--- a/litellm/proxy/db/db_spend_update_writer.py
+++ b/litellm/proxy/db/db_spend_update_writer.py
@@ -7,9 +7,8 @@ from typing import Any, Optional, Union
 import litellm
 from litellm._logging import verbose_proxy_logger
 from litellm.proxy._types import LiteLLM_UserTable, SpendLogsPayload
-from litellm.proxy.proxy_server import hash_token
 from litellm.proxy.spend_tracking.spend_tracking_utils import get_logging_payload
-from litellm.proxy.utils import PrismaClient, ProxyUpdateSpend
+from litellm.proxy.utils import PrismaClient, ProxyUpdateSpend, hash_token
 
 
 class DBSpendUpdateWriter:
diff --git a/tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py b/tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py
index a5ee9ddf70..e08bce432d 100644
--- a/tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py
+++ b/tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py
@@ -416,7 +416,8 @@ class TestSpendLogsPayload:
         # litellm._turn_on_debug()
 
         with patch.object(
-            litellm.proxy.proxy_server, "_set_spend_logs_payload"
+            litellm.proxy.db.db_spend_update_writer.DBSpendUpdateWriter,
+            "_set_spend_logs_payload",
         ) as mock_client, patch.object(litellm.proxy.proxy_server, "prisma_client"):
             response = await litellm.acompletion(
                 model="gpt-4o",
@@ -509,7 +510,8 @@ class TestSpendLogsPayload:
         client = AsyncHTTPHandler()
 
         with patch.object(
-            litellm.proxy.proxy_server, "_set_spend_logs_payload"
+            litellm.proxy.db.db_spend_update_writer.DBSpendUpdateWriter,
+            "_set_spend_logs_payload",
         ) as mock_client, patch.object(
             litellm.proxy.proxy_server, "prisma_client"
         ), patch.object(
@@ -604,7 +606,8 @@ class TestSpendLogsPayload:
         )
 
         with patch.object(
-            litellm.proxy.proxy_server, "_set_spend_logs_payload"
+            litellm.proxy.db.db_spend_update_writer.DBSpendUpdateWriter,
+            "_set_spend_logs_payload",
         ) as mock_client, patch.object(
             litellm.proxy.proxy_server, "prisma_client"
         ), patch.object(