diff --git a/tier_2/deep_learning/Surface_Observation_Prediction_in_Tensorflow.ipynb b/tier_2/deep_learning/Surface_Observation_Prediction_in_Tensorflow.ipynb
index ec98279..99d740d 100644
--- a/tier_2/deep_learning/Surface_Observation_Prediction_in_Tensorflow.ipynb
+++ b/tier_2/deep_learning/Surface_Observation_Prediction_in_Tensorflow.ipynb
@@ -426,7 +426,7 @@
         "\n",
         "    # Then just stack layers on top of one another\n",
         "    # the first specifies the shape of the inputs expected input\n",
-        "    model.add(Input(input_shape, name=\"Inputs\"))\n",
+        "    model.add(Input(shape=[input_shape], name=\"Inputs\"))\n",
         "\n",
         "    # Then we stack on depth number of consectutive dense layers\n",
         "    # To write more compact code we can include the activation\n",
@@ -491,7 +491,7 @@
         "    input_shape: int, width: int, depth: int, activation: str, learning_rate=10**(-3), final_activation: str = \"linear\"\n",
         "):\n",
         "    # First we create an input layer\n",
-        "    inputs = Input(input_shape, name=\"Inputs\")\n",
+        "    inputs = Input(shape=[input_shape], name=\"Inputs\")\n",
         "\n",
         "    # Pass it to the hidden variable we will reuse\n",
         "    hidden = inputs\n",