diff --git a/workflows/ANN_training.smk b/workflows/ANN_training.smk
index 66de3521b5276f94bb58cf87a527eeefab1c18c7..a80af1f206e98d86996bdb773c9d6e98e9a8c88e 100644
--- a/workflows/ANN_training.smk
+++ b/workflows/ANN_training.smk
@@ -47,7 +47,7 @@ rule test_model:
         DIR+'/normalized_input_data.npy',
         DIR+'/output_data.npy'
     output:
-        DIR+'/'+'_'.join(MODELS.keys())+'.json'
+        protected(DIR+'/'+'_'.join(MODELS.keys())+'.json')
     params:
         num_iterations = config['num_iterations'],
         compare_normalization = config['compare_normalization']
@@ -73,8 +73,8 @@ rule train_model:
         DIR+'/normalized_input_data.npy',
         DIR+'/output_data.npy'
     output:
-        DIR+'/trained models/model__{model}.pt',
-        DIR+'/trained models/loss__{model}.pt'
+        protected(DIR+'/trained models/model__{model}.pt'),
+        protected(DIR+'/trained models/loss__{model}.pt')
     log:
         DIR+'/log/train_model_{model}.log'
     run: