diff --git a/06_multicat.ipynb b/06_multicat.ipynb
index 157cd59..bdcf466 100644
--- a/06_multicat.ipynb
+++ b/06_multicat.ipynb
@@ -922,7 +922,7 @@
    ],
    "source": [
     "x,y = to_cpu(dls.train.one_batch())\n",
-    "activs = learn.model(x)\n",
+    "activs = TensorBase(learn.model(x))\n",
     "activs.shape"
    ]
   },