diff --git a/model_prices_and_context_window.schema.json b/model_prices_and_context_window.schema.json
index 6b8ce824f69..1a93cdb504a 100644
--- a/model_prices_and_context_window.schema.json
+++ b/model_prices_and_context_window.schema.json
@@ -314,11 +314,11 @@
           "type": "number",
           "minimum": 0
         },
-        "input_cost_per_reference_pixel": {
+        "input_cost_per_query": {
           "type": "number",
           "minimum": 0
         },
-        "input_cost_per_query": {
+        "input_cost_per_reference_pixel": {
           "type": "number",
           "minimum": 0
         },
diff --git a/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py b/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py
index 608a04959fc..d9f93e19b9f 100644
--- a/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py
+++ b/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py
@@ -180,5 +180,5 @@ def test_unlisted_model_bills_deployment_rates() -> None:
 
 
 def test_non_image_response_raises() -> None:
-    with pytest.raises(ValueError):
+    with pytest.raises(ValueError, match="must be of type ImageResponse"):
         cost_calculator(model="FLUX.2-flex", image_response=object())