diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 76339ee8f4bfffc385d5e84bcac052997133cc79..552d533e043dce1a4d0f7612229f2d2006f894c5 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -1010,7 +1010,7 @@ class IcingIntensityFCN:
         self.do_evaluate(ckpt_dir)
 
 
-def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', use_flight_altitude=False):
+def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', l1b_or_l2='both', use_flight_altitude=False):
     ckpt_dir_s = os.listdir(ckpt_dir_s_path)
     cm_s = []
     prob_s = []
@@ -1020,7 +1020,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA
         ckpt_dir = ckpt_dir_s_path + ckpt
         if not os.path.isdir(ckpt_dir):
             continue
-        nn = IcingIntensityFCN(day_night=day_night, use_flight_altitude=use_flight_altitude)
+        nn = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_or_l2, use_flight_altitude=use_flight_altitude)
         nn.run_restore(filename_l1b, filename_l2, ckpt_dir)
         cm_s.append(tf.math.confusion_matrix(nn.test_labels.flatten(), nn.test_preds.flatten()))
         prob_s.append(nn.test_probs.flatten())