From 89aa08cce7f33e6dafced6f8619f1fc4afdc3c91 Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Tue, 12 Jul 2016 16:02:34 -0400
Subject: [PATCH] simplified function logic

---
 goblin/traversal.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/goblin/traversal.py b/goblin/traversal.py
index 0c2dba3..0f326ea 100644
--- a/goblin/traversal.py
+++ b/goblin/traversal.py
@@ -66,6 +66,7 @@ class Traversal(connection.AbstractConnection):
         return self._session
 
     def traversal(self):
+        traversal = self.graph.traversal()
         if self._element_class:
             label = self._element_class.__mapping__.label
             traversal = self._graph.traversal()
@@ -74,8 +75,6 @@ class Traversal(connection.AbstractConnection):
             if self._element_class.__type__ == 'edge':
                 traversal = traversal.E()
             traversal = traversal.hasLabel(label)
-        else:
-            traversal = self.graph.traversal()
         return traversal
 
     async def submit(self,
-- 
GitLab