diff --git a/src/api/resolvers/Mutation.js b/src/api/resolvers/Mutation.js
index 73b8677c..9427134f 100644
--- a/src/api/resolvers/Mutation.js
+++ b/src/api/resolvers/Mutation.js
@@ -55,13 +55,13 @@ const Mutation = {
   },
 
   updateImageComment: async (_, { input }, context) => {
-    const comment = await context.models.Image.updateComment(input, context);
-    return { comment };
+    const res = await context.models.Image.updateComment(input, context);
+    return { ...res };
   },
 
   createImageComment: async (_, { input }, context) => {
-    const comment = await context.models.Image.createComment(input, context);
-    return { comment };
+    const res = await context.models.Image.createComment(input, context);
+    return { ...res };
   },
 
   createImage: async (_, { input }, context) => {