diff --git a/code/lung_feature_patch_allPatients_singleProtein.py b/code/lung_feature_patch_allPatients_singleProtein.py
index a43234d7a914d4e281bc0ccb0c05bb11c4e45545..9e6a168d8194224f29bbf756ad81fcc36f33f02e 100755
--- a/code/lung_feature_patch_allPatients_singleProtein.py
+++ b/code/lung_feature_patch_allPatients_singleProtein.py
@@ -54,7 +54,7 @@ nr_runs = 10
     
 #%% Define and create directories
 
-disease = 'emphysema' #diseased (mix, 2 of each condition)' #''emphysema' 'sarcoidosis' 'ild' 'covid'
+disease = 'covid' #diseased (mix, 2 of each condition)' #''emphysema' 'sarcoidosis' 'ild' 'covid'
 control = 'control'
 conditions = [disease, control]
     
diff --git a/code/lung_feature_patch_allPatients_twoProtein.py b/code/lung_feature_patch_allPatients_twoProtein.py
index debe53f4e6f8d9a2477c343bd3e71b04bd98bb1b..d8e7c21611098af330107eb68cc3308d4cb2f6b8 100755
--- a/code/lung_feature_patch_allPatients_twoProtein.py
+++ b/code/lung_feature_patch_allPatients_twoProtein.py
@@ -59,7 +59,7 @@ colour_mode = 'cmy'
  
 #%% Define and create directories
 
-disease = 'emphysema' #diseased (mix, 2 of each condition)' #''emphysema' 'sarcoidosis' 'ild' 'covid'
+disease = 'uip' #diseased (mix, 2 of each condition)' #''emphysema' 'sarcoidosis' 'ild' 'covid'
 control = 'control'
 conditions = [disease, control]