diff --git a/src/test/java/com/arangodb/graphql/query/result/resolver/ResultResolverNodeHierarchyTest.java b/src/test/java/com/arangodb/graphql/query/result/resolver/ResultResolverNodeHierarchyTest.java
index 7135c96..efbe408 100644
--- a/src/test/java/com/arangodb/graphql/query/result/resolver/ResultResolverNodeHierarchyTest.java
+++ b/src/test/java/com/arangodb/graphql/query/result/resolver/ResultResolverNodeHierarchyTest.java
@@ -161,6 +161,8 @@ public void processWithTraversal() {
         topLevelFieldValues.put("__collection", "Document");
         topLevelFieldValues.put("secondLevelField", 41);
 
+        when(topLevelFieldDefinition.getDirective("edgeTarget")).thenReturn(null);
+        when(resultVertices.get("Document/1")).thenReturn(null);
         //Have an Edge directive
         when(topLevelFieldDefinition.getDirective("edge")).thenReturn(edgeDirective);
         when(edgeDirective.getArgument("collection")).thenReturn(collectionArgument);