diff --git a/modules/icing/util.py b/modules/icing/util.py
index 8c9e81eca6f94be5d12b282574cc3199f9fededf..2797be110bebb8b290f9e3935b90dfdc289b41dd 100644
--- a/modules/icing/util.py
+++ b/modules/icing/util.py
@@ -319,6 +319,7 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output
                             l1b_andor_l2='BOTH', use_flight_altitude=True, res_fac=1, model_type='CNN',
                             extent=[-105, -70, 15, 50],
                             pirep_file='/Users/tomrink/data/pirep/pireps_202109200000_202109232359.csv',
+                            icing_dict=None,
                             obs_lons=None, obs_lats=None, obs_times=None, obs_alt=None, flight_level=None):
     if model_type == 'CNN':
         model_module = icing_cnn
@@ -332,6 +333,8 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output
 
     if pirep_file is not None:
         ice_dict, no_ice_dict, neg_ice_dict = setup(pirep_file)
+    if icing_dict is not None:
+        ice_dict = icing_dict
 
     alt_lo, alt_hi = 0.0, 15000.0
     if flight_level is not None: