diff --git a/src/core/injectSchemas.ts b/src/core/injectSchemas.ts index a469d68..5f7d2fd 100644 --- a/src/core/injectSchemas.ts +++ b/src/core/injectSchemas.ts @@ -5,7 +5,8 @@ export default function injectSchemas(code: string, refName: string) { const preservedCodeWithSchemasInjected = preservedCode .replace(new RegExp(`\\b${refName}\\.`, "g"), `global.schemas[${refName}].`) - .replace(new RegExp(`\\b${refName}\\b`, "g"), `"${refName}"`); + .replace(new RegExp(`\\b${refName}\\b`, "g"), `"${refName}"`) + .replace(new RegExp(`queryParams:\\s*['"\`]${refName}['"\`]`, "g"), `queryParams: global.schemas["${refName}"]`); return restoreStrings(preservedCodeWithSchemasInjected, replacements); }