diff --git a/notebooks/unit6/unit6.ipynb b/notebooks/unit6/unit6.ipynb
index 95056b5..eb0c370 100644
--- a/notebooks/unit6/unit6.ipynb
+++ b/notebooks/unit6/unit6.ipynb
@@ -5,7 +5,7 @@
     "colab": {
       "provenance": [],
       "private_outputs": true,
-      "authorship_tag": "ABX9TyMm2AvQJHZiNbxotv6J/Rf+",
+      "authorship_tag": "ABX9TyPDFLK3trc6MCLJLqUUuAbl",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -26,7 +26,7 @@
         "colab_type": "text"
       },
       "source": [
-        "<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/ThomasSimonini%2FA2C/notebooks/unit6/unit6.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+        "<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/main/notebooks/unit6/unit6.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
     {
@@ -562,7 +562,7 @@
         "# Load the agent\n",
         "model = A2C.load(\"a2c-AntBulletEnv-v0\")\n",
         "\n",
-        "mean_reward, std_reward = evaluate_policy(model, env)\n",
+        "mean_reward, std_reward = evaluate_policy(model, eval_env)\n",
         "\n",
         "print(f\"Mean reward = {mean_reward:.2f} +/- {std_reward:.2f}\")"
       ],
@@ -862,7 +862,7 @@
         "# Load the agent\n",
         "model = A2C.load(model_name)\n",
         "\n",
-        "mean_reward, std_reward = evaluate_policy(model, env)\n",
+        "mean_reward, std_reward = evaluate_policy(model, eval_env)\n",
         "\n",
         "print(f\"Mean reward = {mean_reward:.2f} +/- {std_reward:.2f}\")\n",
         "\n",