diff --git a/src/graphql-schema.js b/src/graphql-schema.js
index 521ac97bd..ce84dde36 100644
--- a/src/graphql-schema.js
+++ b/src/graphql-schema.js
@@ -1,4 +1,3 @@
-// import { neo4jgraphql } from "neo4j-graphql-js"
 import fs from 'fs'
 import path from 'path'
 import bcrypt from 'bcryptjs'
diff --git a/src/middleware/idMiddleware.js b/src/middleware/idMiddleware.js
index d6dac0580..2f1854fa6 100644
--- a/src/middleware/idMiddleware.js
+++ b/src/middleware/idMiddleware.js
@@ -15,10 +15,14 @@ const includeId = async (resolve, root, args, context, resolveInfo) => {
 }
 
 export default {
-  Query: (resolve, root, args, context, info) => {
-    return includeId(resolve, root, args, context, info)
+  Query: {
+    User: (resolve, root, args, context, info) => {
+      return includeId(resolve, root, args, context, info)
+    }
   },
-  Mutation: (resolve, root, args, context, info) => {
-    return includeId(resolve, root, args, context, info)
+  Mutation: {
+    CreatePost: (resolve, root, args, context, info) => {
+      return includeId(resolve, root, args, context, info)
+    }
   }
 }