diff --git a/site/en/guide/keras/train_and_evaluate.ipynb b/site/en/guide/keras/train_and_evaluate.ipynb
index c8464e937a7..fee9d8f11d9 100644
--- a/site/en/guide/keras/train_and_evaluate.ipynb
+++ b/site/en/guide/keras/train_and_evaluate.ipynb
@@ -519,7 +519,7 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "26a6d8caca22"
+        "id": "e2d7d358b4eb"
       },
       "source": [
         "### Custom metrics\n",
@@ -532,7 +532,7 @@
         "- `update_state(self, y_true, y_pred, sample_weight=None)`, which uses the targets\n",
         "y_true and the model predictions y_pred to update the state variables.\n",
         "- `result(self)`, which uses the state variables to compute the final results.\n",
-        "- `reset_states(self)`, which reinitializes the state of the metric.\n",
+        "- `reset_state(self)`, which reinitializes the state of the metric.\n",
         "\n",
         "State update and results computation are kept separate (in `update_state()` and\n",
         "`result()`, respectively) because in some cases, the results computation might be very\n",
@@ -546,7 +546,7 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "05d6a6e7022d"
+        "id": "2ad9c57c0683"
       },
       "outputs": [],
       "source": [
@@ -567,7 +567,7 @@
         "    def result(self):\n",
         "        return self.true_positives\n",
         "\n",
-        "    def reset_states(self):\n",
+        "    def reset_state(self):\n",
         "        # The state of the metric will be reset at the start of each epoch.\n",
         "        self.true_positives.assign(0.0)\n",
         "\n",