diff --git a/bundle/python_bundle_workflow/scripts/inference.py b/bundle/python_bundle_workflow/scripts/inference.py
index 24810a0dfb..fab321b009 100644
--- a/bundle/python_bundle_workflow/scripts/inference.py
+++ b/bundle/python_bundle_workflow/scripts/inference.py
@@ -63,7 +63,6 @@ def __init__(self, dataset_dir: str = "."):
             nib.save(n, os.path.join(dataset_dir, f"seg{i:d}.nii.gz"))
 
         self._props = {}
-        self._set_props = {}
         self.dataset_dir = dataset_dir
 
     def initialize(self):
@@ -76,7 +75,7 @@ def finalize(self):
         pass
 
     def _set_property(self, name, property, value):
-        self._set_props[name] = value
+        self._props[name] = value
 
     def _get_property(self, name, property):
         """
@@ -90,9 +89,6 @@ def _get_property(self, name, property):
         value = None
         if name in self._props:
             value = self._props[name]
-        elif name in self._set_props:
-            value = self._set_props[name]
-            self._props[name] = value
         else:
             try:
                 value = getattr(self, f"get_{name}")()
diff --git a/bundle/python_bundle_workflow/scripts/train.py b/bundle/python_bundle_workflow/scripts/train.py
index ccd409e969..56fc36056a 100644
--- a/bundle/python_bundle_workflow/scripts/train.py
+++ b/bundle/python_bundle_workflow/scripts/train.py
@@ -72,7 +72,6 @@ def __init__(self, dataset_dir: str = "."):
 
         # define buckets to store the generated properties and set properties
         self._props = {}
-        self._set_props = {}
         self.dataset_dir = dataset_dir
 
         # besides the predefined properties, this bundle workflow can also provide `network`, `loss`, `optimizer`
@@ -91,7 +90,7 @@ def finalize(self):
         set_determinism(None)
 
     def _set_property(self, name, property, value):
-        self._set_props[name] = value
+        self._props[name] = value
 
     def _get_property(self, name, property):
         """
@@ -105,9 +104,6 @@ def _get_property(self, name, property):
         value = None
         if name in self._props:
             value = self._props[name]
-        elif name in self._set_props:
-            value = self._set_props[name]
-            self._props[name] = value
         else:
             try:
                 value = getattr(self, f"get_{name}")()