fix: insert schema keymain

1 files changed, 6 insertions, 2 deletions

diff --git a/builder/builder.go b/builder/builder.go
index 0e90c54..134d7db 100644
--- a/builder/builder.go
+++ b/builder/builder.go
@@ -74,12 +74,16 @@ func WithResponse(status int, mediaType string, schema *jsonschema.Schema) PathO
 
 		if value, ok := responses[strconv.Itoa(status)]; ok {
 			response := value.(map[string]interface{})
-			response["content"].(map[string]interface{})[mediaType] = schema
+			response["content"].(map[string]interface{})[mediaType] = map[string]*jsonschema.Schema{
+				"schema": schema,
+			}
 		} else {
 			responses[strconv.Itoa(status)] = map[string]interface{}{
 				"description": http.StatusText(status),
 				"content": map[string]interface{}{
-					mediaType: schema,
+					mediaType: map[string]*jsonschema.Schema{
+						"schema": schema,
+					},
 				},
 			}
 		}