diff --git a/site/en/tutorials/keras/text_classification.ipynb b/site/en/tutorials/keras/text_classification.ipynb
index c66d0fce0d..02e768d741 100644
--- a/site/en/tutorials/keras/text_classification.ipynb
+++ b/site/en/tutorials/keras/text_classification.ipynb
@@ -861,8 +861,8 @@
         ")\n",
         "\n",
         "# Test it with `raw_test_ds`, which yields raw strings\n",
-        "loss, accuracy = export_model.evaluate(raw_test_ds)\n",
-        "print(accuracy)"
+        "metrics = export_model.evaluate(raw_test_ds, return_dict=True)\n",
+        "print(metrics)"
       ]
     },
     {
@@ -969,7 +969,6 @@
     "accelerator": "GPU",
     "colab": {
       "name": "text_classification.ipynb",
-      "provenance": [],
       "toc_visible": true
     },
     "kernelspec": {