diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py
index ab01f6f19..d1f0ee699 100644
--- a/litellm/tests/test_embedding.py
+++ b/litellm/tests/test_embedding.py
@@ -319,6 +319,12 @@ def test_aembedding_azure():
                     input=["good morning from litellm", "this is another item"],
                 )
                 print(response)
+
+                print(
+                    "hidden params - custom_llm_provider",
+                    response._hidden_params["custom_llm_provider"],
+                )
+                assert response._hidden_params["custom_llm_provider"] == "azure"
             except Exception as e:
                 pytest.fail(f"Error occurred: {e}")