diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index dca0cb67b1bd37ddec3ba9d65e84b2cbd009e604..6c669f7ae6df88c29803c676e337ef45f723c681 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -902,12 +902,10 @@ class IcingIntensityFCN:
 
     def build_model(self):
         cnn = self.build_cnn()
-        # flat_1d = self.build_1d_cnn()
-        # flat = tf.keras.layers.concatenate([flat, flat_1d, flat_anc])
-        # flat = tf.keras.layers.concatenate([flat, flat_1d])
-        # self.build_dnn(flat)
-        ## if self.USE_FLIGHT_ALTITUDE:
-        ##    flat = tf.keras.layers.concatenate([flat, self.inputs[1]])
+        if self.USE_FLIGHT_ALTITUDE:
+            fa = np.expand_dims(self.inputs[1], axis=1)
+            fa = np.expand_dims(fa, axis=1)
+            cnn = tf.keras.layers.concatenate([cnn, fa])
         self.build_fcl(cnn)
         self.model = tf.keras.Model(self.inputs, self.logits)