diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 5b31a587332c1d6054b631e419760219daaa6e6c..63e3d07e9a2c6e50a7ffda8b9f578231950e9223 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -220,8 +220,7 @@ class IcingIntensityFCN:
         self.X_img = tf.keras.Input(shape=(None, None, n_chans))
 
         self.inputs.append(self.X_img)
-        #self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
-        self.inputs.append(tf.keras.Input(shape=(5)))
+        self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
 
         self.flight_level = 0
 
@@ -947,6 +946,7 @@ class IcingIntensityFCN:
 
     def build_model(self):
         cnn = self.build_cnn()
+        print(cnn.shape, self.inputs[1].shape)
         if self.USE_FLIGHT_ALTITUDE:
             cnn = tf.keras.layers.concatenate([cnn, self.inputs[1]])
         self.build_fcl(cnn)