diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 5a6b069fcf9da70bd98d4b8451457bb32c2f0d55..0971b098f098b1325027993c940ef2f7a25ce427 100644
--- a/modules/icing/pirep_goes.py
+++ b/modules/icing/pirep_goes.py
@@ -2178,8 +2178,11 @@ def run_make_images(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', ckpt_dir_s_pat
 def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=homedir,
                       day_model_path=model_path_day, night_model_path=model_path_night,
                       prob_thresh=0.5, satellite='GOES16', domain='CONUS', day_night='AUTO',
-                      l1b_andor_l2='BOTH', use_flight_altitude=True, res_fac=1):
-    flight_levels = [0, 1, 2, 3, 4]
+                      l1b_andor_l2='both', use_flight_altitude=True, res_fac=1):
+    if use_flight_altitude is True:
+        flight_levels = [0, 1, 2, 3, 4]
+    else:
+        flight_levels = [0]
 
     day_train_params = get_training_parameters(day_night='DAY', l1b_andor_l2=l1b_andor_l2)
     nght_train_params = get_training_parameters(day_night='NIGHT', l1b_andor_l2=l1b_andor_l2)
@@ -2311,7 +2314,10 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output
                             pirep_file='/Users/tomrink/data/pirep/pireps_202109200000_202109232359.csv',
                             obs_lons=None, obs_lats=None, obs_times=None, obs_alt=None, flight_level=None):
 
-    flight_levels = [0, 1, 2, 3, 4]
+    if use_flight_altitude is True:
+        flight_levels = [0, 1, 2, 3, 4]
+    else:
+        flight_levels = [0]
 
     if pirep_file is not None:
         ice_dict, no_ice_dict, neg_ice_dict = setup(pirep_file)