diff --git a/src/graphnet/data/__init__.py b/src/graphnet/data/__init__.py
index 512d414d9..742a17b0e 100644
--- a/src/graphnet/data/__init__.py
+++ b/src/graphnet/data/__init__.py
@@ -8,4 +8,9 @@
 from .pre_configured import I3ToParquetConverter
 from .pre_configured import I3ToSQLiteConverter
 from .datamodule import GraphNeTDataModule
-from .curated_datamodule import CuratedDataset, ERDAHostedDataset
+from .curated_datamodule import (
+    CuratedDataset,
+    ERDAHostedDataset,
+    SecretDataset,
+    PublicBenchmarkDataset,
+)