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 798007b829..f8b8c53161 100644
--- a/tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py
+++ b/tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py
@@ -15,6 +15,7 @@ sys.path.insert(
 from unittest.mock import MagicMock, patch
 
 import litellm
+import litellm_proxy
 from litellm.router import Router
 from litellm_proxy._types import SpendLogsPayload
 from litellm_proxy.hooks.proxy_track_cost_callback import _ProxyDBLogger