diff --git a/course/en/chapter3/section3_tf.ipynb b/course/en/chapter3/section3_tf.ipynb
index cd1f4cf6..396e3034 100644
--- a/course/en/chapter3/section3_tf.ipynb
+++ b/course/en/chapter3/section3_tf.ipynb
@@ -20,7 +20,19 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install datasets evaluate transformers[sentencepiece]"
+    "!pip install datasets evaluate transformers[sentencepiece]\n",
+    "!pip install --upgrade tensorflow\n",
+    "!pip install --upgrade tf-keras"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "os.environ['TF_USE_LEGACY_KERAS'] = '1'"
    ]
   },
   {
@@ -195,6 +207,9 @@
   "colab": {
    "name": "Fine-tuning a model with the Trainer API or Keras",
    "provenance": []
+  },
+  "language_info": {
+   "name": "python"
   }
  },
  "nbformat": 4,