diff --git a/tests/test_model_registry_schemas.py b/tests/test_model_registry_schemas.py
index d9a840d..88c6d8e 100644
--- a/tests/test_model_registry_schemas.py
+++ b/tests/test_model_registry_schemas.py
@@ -68,6 +68,7 @@ ALL_ERROR_CODES = {
     "MODEL_CAPABILITY_UNAVAILABLE": 422,
     "UNSUPPORTED_RUNTIME_PARAMETER": 422,
     "MODEL_LIMITS_UNCONFIRMED": 422,
+    "IMAGE_MODEL_NOT_CHAT_MODEL": 422,
     "CONTEXT_BUDGET_UNSATISFIABLE": 422,
     "PROVIDER_UNREACHABLE": 422,
     "VALIDATION_FAILED": 422,