diff --git a/docs/get_started.ipynb b/docs/get_started.ipynb
index 5dcf8f6e46..82cde1e778 100644
--- a/docs/get_started.ipynb
+++ b/docs/get_started.ipynb
@@ -135,7 +135,8 @@
         "\n",
         "def create_model():\n",
         "  return tf.keras.models.Sequential([\n",
-        "    tf.keras.layers.Flatten(input_shape=(28, 28), name='layers_flatten'),\n",
+        "    tf.keras.layers.Input(shape=(28, 28), name='layers_input'),\n",
+        "    tf.keras.layers.Flatten(name='layers_flatten'),\n",
         "    tf.keras.layers.Dense(512, activation='relu', name='layers_dense'),\n",
         "    tf.keras.layers.Dropout(0.2, name='layers_dropout'),\n",
         "    tf.keras.layers.Dense(10, activation='softmax', name='layers_dense_2')\n",
@@ -452,10 +453,10 @@
         "                         test_accuracy.result()*100))\n",
         "\n",
         "  # Reset metrics every epoch\n",
-        "  train_loss.reset_states()\n",
-        "  test_loss.reset_states()\n",
-        "  train_accuracy.reset_states()\n",
-        "  test_accuracy.reset_states()"
+        "  train_loss.reset_state()\n",
+        "  test_loss.reset_state()\n",
+        "  train_accuracy.reset_state()\n",
+        "  test_accuracy.reset_state()"
       ]
     },
     {