diff --git a/pts/dataset/__init__.py b/pts/dataset/__init__.py
index 82a9193..6ce8cb7 100644
--- a/pts/dataset/__init__.py
+++ b/pts/dataset/__init__.py
@@ -9,6 +9,6 @@ from .sampler import (
     UniformSplitSampler,
 )
 from .process import ProcessStartField, ProcessDataEntry
-from .utils import to_pandas, 
+from .utils import to_pandas
 from .stat import ScaleHistogram, calculate_dataset_statistics
 from .artificial import constant_dataset
\ No newline at end of file
diff --git a/pts/model/__init__.py b/pts/model/__init__.py
index 6c87161..7287791 100644
--- a/pts/model/__init__.py
+++ b/pts/model/__init__.py
@@ -1,5 +1,5 @@
 from .estimator import Estimator, PTSEstimator
-from .forecast import Forecast, SampleForecast, QuantileForecast
+from .forecast import Forecast, SampleForecast, QuantileForecast, DistributionForecast
 from .predictor import Predictor
 from .quantile import Quantile
 from .utils import get_module_forward_input_names, copy_parameters
\ No newline at end of file