diff --git a/notebooks/differentiable-parameterizations/xy2rgb.ipynb b/notebooks/differentiable-parameterizations/xy2rgb.ipynb
index 2f2fdc9b..57210e58 100644
--- a/notebooks/differentiable-parameterizations/xy2rgb.ipynb
+++ b/notebooks/differentiable-parameterizations/xy2rgb.ipynb
@@ -103,6 +103,15 @@
         "## Install, Import, and load a model"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "%tensorflow_version 1.x"
+      ]
+    },
     {
       "metadata": {
         "id": "b-IocmWWvb_I",
@@ -1116,4 +1125,4 @@
       ]
     }
   ]
-}
+}
\ No newline at end of file