diff --git a/packages/amplify-graphql-generation-transformer/src/__tests__/__snapshots__/amplify-graphql-generation-transformer.test.ts.snap b/packages/amplify-graphql-generation-transformer/src/__tests__/__snapshots__/amplify-graphql-generation-transformer.test.ts.snap
index 253d70f261..267576cd34 100644
--- a/packages/amplify-graphql-generation-transformer/src/__tests__/__snapshots__/amplify-graphql-generation-transformer.test.ts.snap
+++ b/packages/amplify-graphql-generation-transformer/src/__tests__/__snapshots__/amplify-graphql-generation-transformer.test.ts.snap
@@ -304,6 +304,10 @@ export function response(ctx) {
   }
 
   const response = toolUse.input.value;
+  if (typeof response === 'string') {
+    return JSON.parse(response);
+  }
+
   return response;
 }"
 `;
@@ -404,6 +408,10 @@ export function response(ctx) {
   }
 
   const response = toolUse.input.value;
+  if (typeof response === 'string') {
+    return JSON.parse(response);
+  }
+
   return response;
 }"
 `;