diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index c2539a9ae4a59911fa1ffd4f5b9476cd0c899b64..9002f2902b2a5151e31dd00c32e6a8300b4852f3 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -219,7 +219,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(None, None, 5))
+        self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
 
         self.flight_level = 0