diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 5388ef3432b0c90c2348a15d92705ab03c2ed09c..8a63918997c15af5e4e9d6d380d67a94487955d1 100644
--- a/modules/icing/pirep_goes.py
+++ b/modules/icing/pirep_goes.py
@@ -2296,13 +2296,12 @@ def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=h
         for flvl in flight_levels:
             fd_preds = preds_2d_dct[flvl]
             fd_probs = probs_2d_dct[flvl]
-            if use_nan:
-                fd_probs = np.where(fd_probs < prob_thresh, np.nan, fd_probs)
             preds_2d_dct[flvl] = fd_preds.reshape((num_lines, num_elems))
             probs_2d_dct[flvl] = fd_probs.reshape((num_lines, num_elems))
 
         write_icing_file_nc4(clvrx_str_time, output_dir, preds_2d_dct, probs_2d_dct,
-                             x_rad, y_rad, lons_2d, lats_2d, cc, ll, satellite=satellite, domain=domain, use_nan=use_nan)
+                             x_rad, y_rad, lons_2d, lats_2d, cc, ll,
+                             satellite=satellite, domain=domain, use_nan=use_nan, prob_thresh=prob_thresh)
 
         print('Done: ', clvrx_str_time)
         h5f.close()