diff --git a/datamint/__init__.py b/datamint/__init__.py
index 00578c3..9ba4915 100644
--- a/datamint/__init__.py
+++ b/datamint/__init__.py
@@ -11,9 +11,17 @@
 if TYPE_CHECKING:
     from .api.client import Api as Api
 
-    # New modular datasets
+    from .dataset.base import DatamintBaseDataset as DatamintBaseDataset
+    from .dataset.base import DatamintDatasetException as DatamintDatasetException
+    from .dataset.factory import build_dataset as build_dataset
     from .dataset.image_dataset import ImageDataset as ImageDataset
+    from .dataset.multiframe_dataset import MultiFrameDataset as MultiFrameDataset
+    from .dataset.sliced_dataset import SlicedVolumeDataset as SlicedVolumeDataset
+    from .dataset.sliced_video_dataset import SlicedVideoDataset as SlicedVideoDataset
+    from .dataset.split_result import SplitResult as SplitResult
+    from .dataset.video_dataset import VideoDataset as VideoDataset
     from .dataset.volume_dataset import VolumeDataset as VolumeDataset
+
     from .default_project import select_project as select_project
     from .importers.coco import COCOImporter as COCOImporter
     from .importers.pascal_voc import PascalVOCImporter as PascalVOCImporter
@@ -34,7 +42,14 @@
         submod_attrs={
             "api.client": ["Api"],
             # New modular dataset classes
+            "dataset.base": ["DatamintBaseDataset", "DatamintDatasetException"],
+            "dataset.factory": ["build_dataset"],
             "dataset.image_dataset": ["ImageDataset"],
+            "dataset.multiframe_dataset": ["MultiFrameDataset"],
+            "dataset.sliced_dataset": ["SlicedVolumeDataset"],
+            "dataset.sliced_video_dataset": ["SlicedVideoDataset"],
+            "dataset.split_result": ["SplitResult"],
+            "dataset.video_dataset": ["VideoDataset"],
             "dataset.volume_dataset": ["VolumeDataset"],
             "mlflow.flavors.validation": ["validate_model", "ValidationReport",
                                           "ValidationIssue", "ModelValidationError"],