diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index ac97090cfae90085854bf42a273abf14bc36d013..37746e4409e1646b1c29aba0a2e725c599a8f5cc 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -151,7 +151,7 @@ class IcingIntensityFCN:
         #     elif l1b_or_l2 == 'l2':
         #         self.train_params = train_params_l2_night
 
-        self.train_params = get_training_parameters(day_night=day_night, l1b_andor_l2=l1b_or_l2, satellite=satellite)
+        self.train_params, self.train_params_l1b, self.train_params_l2 = get_training_parameters(day_night=day_night, l1b_andor_l2=l1b_or_l2, satellite=satellite)
 
         self.train_data = None
         self.train_label = None