diff --git a/pts/dataset/__init__.py b/pts/dataset/__init__.py
index 10171b7..a16a9cf 100644
--- a/pts/dataset/__init__.py
+++ b/pts/dataset/__init__.py
@@ -1,5 +1,6 @@
-from .common import DataEntry, FieldName, Dataset
+from .common import DataEntry, FieldName, Dataset, MetaData, TrainDatasets
 from .list_dataset import ListDataset
+from .file_dataset import FileDataset
 from .loader import TrainDataLoader, InferenceDataLoader
 from .process import ProcessStartField, ProcessDataEntry
 from .utils import to_pandas