diff --git a/litellm/main.py b/litellm/main.py
index a12f8706e..64dcbe49a 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -27,7 +27,8 @@ from litellm.utils import (
     completion_with_fallbacks,
     get_llm_provider,
     get_api_key,
-    mock_completion_streaming_obj
+    mock_completion_streaming_obj,
+    batch_completion_rate_limits
 )
 from .llms import anthropic
 from .llms import together_ai