diff --git a/mlair/run_modules/experiment_setup.py b/mlair/run_modules/experiment_setup.py
index 18f959822e11856a26e87167e9cce17a49bdf47d..0f88fe7395e2e72e72232585ac7a4540f8cdcd4d 100644
--- a/mlair/run_modules/experiment_setup.py
+++ b/mlair/run_modules/experiment_setup.py
@@ -70,6 +70,8 @@ class ExperimentSetup(RunEnvironment):
         * `target_dim` [.]
         * `window_lead_time` [.]
         * `num_realizations` [postprocessing]
+        * `ens_moment_dim` [postprocessing]
+        * `ens_realization_dim` [postprocessing]
 
     Creates
         * plot of model architecture in `<model_name>.pdf`
@@ -205,7 +207,7 @@ class ExperimentSetup(RunEnvironment):
         is not provided at all, no snapshot is loaded. Note, the workflow will apply the default preprocessing without
         loading a snapshot only if this parameter is None!
     :param num_realizations: Number of realizations to be applied using a probabilistic neural network
-    :param ens_para_dim: Name of dimension used to extract ensemble parameters
+    :param ens_moment_dim: Name of dimension used to extract ensemble moments
     :param ens_realization_dim: Name of dimension used for realizations of ensemble
     """