diff --git a/src/graphnet/models/graphs/nodes/__init__.py b/src/graphnet/models/graphs/nodes/__init__.py
index 0119d2b98..1fbafd43f 100644
--- a/src/graphnet/models/graphs/nodes/__init__.py
+++ b/src/graphnet/models/graphs/nodes/__init__.py
@@ -5,4 +5,9 @@
 and their features.
 """
 
-from .nodes import NodeDefinition, NodesAsPulses, PercentileClusters
+from .nodes import (
+    NodeDefinition,
+    NodesAsPulses,
+    PercentileClusters,
+    NodeAsDOMTimeSeries,
+)