diff --git a/tests/llm_translation/test_max_completion_tokens.py b/tests/llm_translation/test_max_completion_tokens.py
index d44a2db37..53f9fc761 100644
--- a/tests/llm_translation/test_max_completion_tokens.py
+++ b/tests/llm_translation/test_max_completion_tokens.py
@@ -235,7 +235,7 @@ def test_all_model_configs():
         drop_params=False,
     ) == {"max_tokens": 10}
 
-    from litellm.llms.bedrock.chat import AmazonConverseConfig
+    from litellm.llms.bedrock.chat.converse_transformation import AmazonConverseConfig
 
     assert (
         "max_completion_tokens"