diff --git a/ci/build.sh b/ci/build.sh
index 7c43de460fe5ad1e9b5eeb593feb9c1a9db145f3..f29f50f27b88056216e6f880bb713012fc4e9956 100755
--- a/ci/build.sh
+++ b/ci/build.sh
@@ -5,7 +5,6 @@
 # Will automatically login to a registry if CI_REGISTRY, CI_REGISTRY_USER and CI_REGISTRY_PASSWORD are set.
 # Will only push an image if $CI_REGISTRY is set.
 
-VERSION="POC"
 if [ -z "$VERSION" ]; then
 	VERSION=${CI_COMMIT_TAG:-latest}
 fi
diff --git a/worker_generic_training_dataset/worker.py b/worker_generic_training_dataset/worker.py
index 7c20093aef13364a921299db91b3144c9130088f..f547e42b86574943bd3c422951b2ab3f88f91374 100644
--- a/worker_generic_training_dataset/worker.py
+++ b/worker_generic_training_dataset/worker.py
@@ -77,15 +77,15 @@ class DatasetExtractor(BaseWorker):
         """
         logger.info("Retrieving information from process_information")
 
-        train_folder_id = self.config.get("train_folder_id")
+        train_folder_id = self.process_information.get("train_folder_id")
         assert train_folder_id, "A training folder id is necessary to use this worker"
         self.training_folder_id = UUID(train_folder_id)
 
-        val_folder_id = self.config.get("validation_folder_id")
+        val_folder_id = self.process_information.get("validation_folder_id")
         assert val_folder_id, "A validation folder id is necessary to use this worker"
         self.validation_folder_id = UUID(val_folder_id)
 
-        test_folder_id = self.config.get("test_folder_id")
+        test_folder_id = self.process_information.get("test_folder_id")
         self.testing_folder_id: UUID | None = (
             UUID(test_folder_id) if test_folder_id else None
         )