diff --git a/yaramo/model.py b/yaramo/model.py
index ee936c8..ec879fa 100644
--- a/yaramo/model.py
+++ b/yaramo/model.py
@@ -1,6 +1,6 @@
 from yaramo.edge import Edge  # pylint: noqa
 from yaramo.geo_node import GeoNode, Wgs84GeoNode, DbrefGeoNode  # pylint: noqa
-from yaramo.node import Node  # pylint: noqa
+from yaramo.node import Node, NodeConnectionDirection  # pylint: noqa
 from yaramo.route import Route  # pylint: noqa
 from yaramo.signal import Signal, SignalDirection, SignalFunction, SignalKind  # pylint: noqa
 from yaramo.topology import Topology  # pylint: noqa