diff --git a/goblin/traversal.py b/goblin/traversal.py index 0c2dba305c8cb5fcf77361fcbc71e81ff93fe81b..0f326ea01bfeb4674f20d5d0c689a91365e2c94a 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,