diff --git a/src/graphnet/models/graphs/__init__.py b/src/graphnet/models/graphs/__init__.py
index 77387b607..e6970bc62 100644
--- a/src/graphnet/models/graphs/__init__.py
+++ b/src/graphnet/models/graphs/__init__.py
@@ -11,5 +11,4 @@
     EdgelessGraph,
     KNNGraphRRWP,
     KNNGraphRWSE,
-    KNNGraphNoPE,
 )