diff --git a/ml/cc/exercises/representation_with_a_feature_cross.ipynb b/ml/cc/exercises/representation_with_a_feature_cross.ipynb
index 23f9f94..2f51559 100644
--- a/ml/cc/exercises/representation_with_a_feature_cross.ipynb
+++ b/ml/cc/exercises/representation_with_a_feature_cross.ipynb
@@ -82,26 +82,34 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "xchnxAsaKKqO",
-        "colab_type": "text"
+        "id": "xchnxAsaKKqO"
       },
       "source": [
         "## Use the right version of TensorFlow\n",
         "\n"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "UZhrUFOntMAt"
+      },
+      "source": [
+        "The following hidden code cell ensures that the Colab will run on TensorFlow 2.X."
+      ]
+    },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "qBpGi_GD14-p",
-        "colab_type": "code",
-        "colab": {}
+        "cellView": "form",
+        "id": "zm_5iTV8tOyl"
       },
+      "outputs": [],
       "source": [
-        "%tensorflow_version 2.x"
-      ],
-      "execution_count": 0,
-      "outputs": []
+        "#@title Run on TensorFlow 2.x\n",
+        "%tensorflow_version 2.x\n"
+      ]
     },
     {
       "cell_type": "markdown",