diff --git a/README.rst b/README.rst
index e8d81afb0504b5a3be230a4b93edee147f6fe004..097f436893f9b02bff1219e69aa6b728d7d262de 100644
--- a/README.rst
+++ b/README.rst
@@ -2,9 +2,9 @@
 .. Mon 03 Jul 2017 11:50:06 CEST
 
 .. image:: http://img.shields.io/badge/docs-stable-yellow.svg
-   :target: http://beatubulatest.lab.idiap.ch/private/docs/bob/bob.paper.mccnn.tifs2018/stable/index.html
+   :target: http://idiap.ch/software/docs/bob/bob.paper.mccnn.tifs2018/stable/index.html
 .. image:: http://img.shields.io/badge/docs-latest-orange.svg
-   :target: http://beatubulatest.lab.idiap.ch/private/docs/bob/bob.paper.mccnn.tifs2018/master/index.html
+   :target: http://idiap.ch/software/docs/bob/bob.paper.mccnn.tifs2018/master/index.html
 .. image:: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/badges/master/build.svg
    :target: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/commits/master
 .. image:: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/badges/master/coverage.svg
@@ -80,4 +80,4 @@ development `mailing list`_.
 .. _mailing list: https://www.idiap.ch/software/bob/discuss
 .. _bob package development: https://www.idiap.ch/software/bob/docs/bob/bob.extension/master/
 .. _conda: https://conda.io
-.. _install conda: https://conda.io/docs/install/quick.html#linux-miniconda-install
\ No newline at end of file
+.. _install conda: https://conda.io/docs/install/quick.html#linux-miniconda-install
diff --git a/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py b/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py
index 1ac899f3b06c2639066bf151d6a1ed9e606bf597..015c32287f1441635a9ebca5b18e1f314b85a06e 100644
--- a/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py
+++ b/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py
@@ -70,7 +70,9 @@ SELECTED_CHANNELS = [0, 1, 2, 3]
 
 img_transform_train = transforms.Compose([ChannelSelect(selected_channels=SELECTED_CHANNELS), RandomHorizontalFlipImage(p=0.5), transforms.ToTensor()])
 
-dataset = DataFolder(data_folder=data_folder_train,
+dataset={}
+
+dataset['train'] = DataFolder(data_folder=data_folder_train,
                      transform=img_transform_train,
                      extension='.hdf5',
                      bob_hldi_instance=bob_hldi_instance_train,