diff --git a/graphene_mongo/tests/models.py b/graphene_mongo/tests/models.py
index 803ca520..ad0b03ca 100644
--- a/graphene_mongo/tests/models.py
+++ b/graphene_mongo/tests/models.py
@@ -8,9 +8,6 @@
 )
 
 
-# mongoengine.connect('graphene-mongo-test', host='mongodb://localhost/graphene-mongo-dev')
-
-
 class Publisher(mongoengine.Document):
     meta = {"collection": "test_publisher"}
     name = mongoengine.StringField()
diff --git a/graphene_mongo/types.py b/graphene_mongo/types.py
index 20ddbf5b..c77cc09a 100644
--- a/graphene_mongo/types.py
+++ b/graphene_mongo/types.py
@@ -22,10 +22,8 @@ def construct_fields(model, registry, only_fields, exclude_fields, non_required_
         registry (.registry.Registry):
         only_fields ([str]):
         exclude_fields ([str]):
-
     Returns:
         (OrderedDict, OrderedDict): converted fields and self reference fields.
-
     """
     _model_fields = get_model_fields(model)
     fields = OrderedDict()
@@ -239,6 +237,9 @@ def get_node(cls, info, id):
         def resolve_id(self, info):
             return str(self.id)
 
+        def __resolve_reference(self, info):
+            return self._meta.model.objects.get(pk=self.id)
+
     return GrapheneMongoengineGenericType, MongoengineGenericObjectTypeOptions