diff --git a/10_neural_nets_with_keras.ipynb b/10_neural_nets_with_keras.ipynb
index 5dd3eb0..6a3710a 100644
--- a/10_neural_nets_with_keras.ipynb
+++ b/10_neural_nets_with_keras.ipynb
@@ -1569,7 +1569,7 @@
     "normalized = normalization_layer(input_)\n",
     "hidden1 = hidden_layer1(normalized)\n",
     "hidden2 = hidden_layer2(hidden1)\n",
-    "concat = concat_layer([input_, hidden2])\n",
+    "concat = concat_layer([normalized, hidden2])\n",
     "output = output_layer(concat)\n",
     "\n",
     "model = tf.keras.Model(inputs=[input_], outputs=[output])"