diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index c5c23f86e29bb4a960ce8e3b20f9f9669c1e2fe5..5a6b069fcf9da70bd98d4b8451457bb32c2f0d55 100644
--- a/modules/icing/pirep_goes.py
+++ b/modules/icing/pirep_goes.py
@@ -2075,7 +2075,7 @@ def analyze(preds_file, test_file='/Users/tomrink/data/icing_ml/tiles_2021092400
     print(np.nanmean(cld_dz[(nda == 4) & true_no_ice]), np.nanmean(cld_hgt[(nda == 4) & true_no_ice]))
 
 
-def get_training_parameters(day_night='DAY', l1b_andor_l2='BOTH'):
+def get_training_parameters(day_night='DAY', l1b_andor_l2='both'):
     if day_night == 'DAY':
         train_params_l2 = ['cld_height_acha', 'cld_geo_thick', 'cld_temp_acha', 'cld_press_acha', 'supercooled_cloud_fraction',
                            'cld_emiss_acha', 'conv_cloud_fraction', 'cld_reff_dcomp', 'cld_opd_dcomp', 'iwc_dcomp', 'lwc_dcomp']
@@ -2090,7 +2090,7 @@ def get_training_parameters(day_night='DAY', l1b_andor_l2='BOTH'):
         train_params_l1b = ['temp_10_4um_nom', 'temp_11_0um_nom', 'temp_12_0um_nom', 'temp_13_3um_nom', 'temp_3_75um_nom',
                             'temp_6_2um_nom', 'temp_6_7um_nom', 'temp_7_3um_nom', 'temp_8_5um_nom', 'temp_9_7um_nom']
 
-    if l1b_andor_l2 == 'BOTH':
+    if l1b_andor_l2 == 'both':
         train_params = train_params_l1b + train_params_l2
     elif l1b_andor_l2 == 'l1b':
         train_params = train_params_l1b