diff --git a/README.md b/README.md
index c24ea7a6..06b88572 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@
 [![shield_pypi_downloads](https://img.shields.io/pypi/dm/aucmedi?style=flat-square)](https://pypistats.org/packages/aucmedi)
 [![shield_license](https://img.shields.io/github/license/frankkramer-lab/aucmedi?style=flat-square)](https://www.gnu.org/licenses/gpl-3.0.en.html)
 
-The open-source software AUCMEDI allows fast setup of medical image classification pipelines with state-of-the-art methods via an intuitive, high-level Python API or via an AutoML deployment through Docker/CLI .
+The open-source software AUCMEDI allows fast setup of medical image classification pipelines with state-of-the-art methods via an intuitive, high-level Python API or via an AutoML deployment through Docker/CLI.
 
 ## Work in Progress!
 
diff --git a/aucmedi/ensemble/augmenting.py b/aucmedi/ensemble/augmenting.py
index f50f3442..bda7de60 100644
--- a/aucmedi/ensemble/augmenting.py
+++ b/aucmedi/ensemble/augmenting.py
@@ -103,6 +103,7 @@ def predict_augmenting(model, prediction_generator, n_cycles=10, aggregate="mean
     aug_gen = DataGenerator(samples_aug,
                             path_imagedir=prediction_generator.path_imagedir,
                             labels=None,
+                            metadata=prediction_generator.metadata,
                             batch_size=prediction_generator.batch_size,
                             data_aug=data_aug,
                             seed=prediction_generator.seed,
diff --git a/docs/README.PyPI.md b/docs/README.PyPI.md
index 113f51cc..691d3f92 100644
--- a/docs/README.PyPI.md
+++ b/docs/README.PyPI.md
@@ -8,7 +8,7 @@
 [![shield_pypi_downloads](https://img.shields.io/pypi/dm/aucmedi?style=flat-square)](https://pypistats.org/packages/aucmedi)
 [![shield_license](https://img.shields.io/github/license/frankkramer-lab/aucmedi?style=flat-square)](https://www.gnu.org/licenses/gpl-3.0.en.html)
 
-The open-source software AUCMEDI allows fast setup of medical image classification pipelines with state-of-the-art methods via an intuitive, high-level Python API or via an AutoML deployment through Docker/CLI .
+The open-source software AUCMEDI allows fast setup of medical image classification pipelines with state-of-the-art methods via an intuitive, high-level Python API or via an AutoML deployment through Docker/CLI.
 
 ## Resources
 - Website: [AUCMEDI Website - Home](https://frankkramer-lab.github.io/aucmedi/)
diff --git a/setup.py b/setup.py
index 1efc4459..b3c48761 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name='aucmedi',
-    version='0.5.1',
+    version='0.5.2',
     description='AUCMEDI - a framework for Automated Classification of Medical Images',
     author='Dominik Müller',
     author_email='dominik.mueller@informatik.uni-augsburg.de',