diff --git a/doc/source/examples/alibi_detect_deploy.ipynb b/doc/source/examples/alibi_detect_deploy.ipynb
index 98e3a9e77..d9ffceadc 100644
--- a/doc/source/examples/alibi_detect_deploy.ipynb
+++ b/doc/source/examples/alibi_detect_deploy.ipynb
@@ -26,7 +26,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {
     "code_folding": [
      0
@@ -56,6 +56,9 @@
    ],
    "source": [
     "# imports and plot examples\n",
+    "import os\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"1\"\n",
+    "\n",
     "import matplotlib.pyplot as plt\n",
     "%matplotlib inline\n",
     "import tensorflow as tf\n",
diff --git a/docs-gb/examples/alibi_detect_deploy.ipynb b/docs-gb/examples/alibi_detect_deploy.ipynb
index 98e3a9e77..d9ffceadc 100644
--- a/docs-gb/examples/alibi_detect_deploy.ipynb
+++ b/docs-gb/examples/alibi_detect_deploy.ipynb
@@ -26,7 +26,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {
     "code_folding": [
      0
@@ -56,6 +56,9 @@
    ],
    "source": [
     "# imports and plot examples\n",
+    "import os\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"1\"\n",
+    "\n",
     "import matplotlib.pyplot as plt\n",
     "%matplotlib inline\n",
     "import tensorflow as tf\n",