diff --git a/tensorflow_ranking/examples/keras/landingpage_example.ipynb b/tensorflow_ranking/examples/keras/landingpage_example.ipynb
index bf91964..9843ff8 100644
--- a/tensorflow_ranking/examples/keras/landingpage_example.ipynb
+++ b/tensorflow_ranking/examples/keras/landingpage_example.ipynb
@@ -43,7 +43,7 @@
         "    \u003ca target=\"_blank\" href=\"https://www.tensorflow.org/ranking/overview\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/tf_logo_32px.png\" /\u003eView on TensorFlow.org\u003c/a\u003e\n",
         "  \u003c/td\u003e\n",
         "  \u003ctd\u003e\n",
-        "    \u003ca target=\"_blank\" href=\"https://github.com/tensorflow/ranking/blob/master/examples/keras/landingpage_example.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" /\u003eView source on GitHub\u003c/a\u003e\n",
+        "    \u003ca target=\"_blank\" href=\"https://github.com/tensorflow/ranking/blob/master/tensorflow_ranking/examples/keras/landingpage_example.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" /\u003eView source on GitHub\u003c/a\u003e\n",
         "  \u003c/td\u003e\n",
         "\u003c/table\u003e"
       ]
@@ -101,7 +101,7 @@
         "\n",
         "# Create a model with Keras Functional API.\n",
         "inputs = {\n",
-        "    name: tf.keras.Input(shape=(None, 1), dtype=tf.float32, name=name)\n",
+        "    name: tf.keras.Input(shape=(None, 136), dtype=tf.float32, name=name)\n",
         "    for name in ds.element_spec[0]\n",
         "    if name != \"_mask\"\n",
         "}\n",