diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index 7913451..4de088f 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -70,7 +70,8 @@
         "torch = True #@param {type:\"boolean\"}\n",
         "reusable = False #@param {type:\"boolean\"}\n",
         "model_class = MinDalleTorch if torch else MinDalleFlax\n",
-        "model = model_class(mega, not expendable)\n"
+        "is_expendable = not reusable\n",
+        "model = model_class(mega, is_expendable)\n"
       ],
       "metadata": {
         "id": "8W-L2ICFGFup",
@@ -79,7 +80,7 @@
           "base_uri": "https://localhost:8080/"
         }
       },
-      "execution_count": 6,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -105,7 +106,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 7,
+      "execution_count": null,
       "metadata": {
         "colab": {
           "base_uri": "https://localhost:8080/",
@@ -172,7 +173,7 @@
       ],
       "name": "min-dalle",
       "provenance": [],
-      "authorship_tag": "ABX9TyMtWeoRGqaMmLjAXcNCf4AW",
+      "authorship_tag": "ABX9TyPWA14jCgKCRzp2RXXzxZ+q",
       "include_colab_link": true
     },
     "gpuClass": "standard",