diff --git a/src/test/java/com/arangodb/graphql/create/DatabaseObjectCreatorTest.java b/src/test/java/com/arangodb/graphql/create/DatabaseObjectCreatorTest.java
index e278c40..83c2b4e 100644
--- a/src/test/java/com/arangodb/graphql/create/DatabaseObjectCreatorTest.java
+++ b/src/test/java/com/arangodb/graphql/create/DatabaseObjectCreatorTest.java
@@ -89,7 +89,7 @@ private void attachVertexDirectiveToMockType() {
 
     private void attachEdgeDirectiveToMockType() {
         attachVertexDirectiveToMockType();
-
+        when(mockEdgeDirective.getArgument("direction")).thenReturn(null);
         when(mockFieldDefinition.getDirective("edge")).thenReturn(mockEdgeDirective);
         when(mockEdgeDirective.getArgument("collection")).thenReturn(mockEdgeArgument);
         when(mockEdgeArgument.getValue()).thenReturn(edgeCollectionName);