diff --git a/DLWMLS/__main__.py b/DLWMLS/__main__.py
index 8c5bf07..39652b2 100644
--- a/DLWMLS/__main__.py
+++ b/DLWMLS/__main__.py
@@ -258,9 +258,10 @@ def main() -> None:
     model_folder = os.path.join(
         Path(__file__).parent,
         "nnunet_results",
-        "Dataset%s_Dataset%s_dlwmlsICVAll/nnUNetTrainer__nnUNetPlans__%s"
+        "Dataset004_Dataset003_qcedCorrected/nnUNetTrainer__nnUNetPlans__3d_fullres"
+        # "Dataset%s_Dataset%s_dlwmlsICVAll/nnUNetTrainer__nnUNetPlans__%s"
         # "Dataset%s_Task%s_DLWMLSV2/nnUNetTrainer__nnUNetPlans__%s/"
-        % (args.d, args.d, args.c),
+        # % (args.d, args.d, args.c),
     )
 
 
diff --git a/DLWMLS/__pycache__/__main__.cpython-312.pyc b/DLWMLS/__pycache__/__main__.cpython-312.pyc
index 3890dec..5215399 100644
Binary files a/DLWMLS/__pycache__/__main__.cpython-312.pyc and b/DLWMLS/__pycache__/__main__.cpython-312.pyc differ
diff --git a/DLWMLS/__pycache__/utils.cpython-312.pyc b/DLWMLS/__pycache__/utils.cpython-312.pyc
index fc10d70..25f5c04 100644
Binary files a/DLWMLS/__pycache__/utils.cpython-312.pyc and b/DLWMLS/__pycache__/utils.cpython-312.pyc differ