diff --git a/pipelines.py b/pipelines.py
index 97d4d57..ae36a0b 100644
--- a/pipelines.py
+++ b/pipelines.py
@@ -9,12 +9,8 @@
 
 
 def unet(config, train_mode):
-    if train_mode:
-        save_output = False
-        load_saved_output = False
-    else:
-        save_output = False
-        load_saved_output = False
+    save_output = False
+    load_saved_output = False
 
     loader = preprocessing(config, model_type='single', is_train=train_mode)
     unet = Step(name='unet',