diff --git a/jupyter/alexnet_insights.ipynb b/jupyter/alexnet_insights.ipynb
index bedaaf0b50fd8ea17984afed6b5c9ad26fd2d4d0..fec514bba46015b048338f0bc4db2f91c0f79ee7 100644
--- a/jupyter/alexnet_insights.ipynb
+++ b/jupyter/alexnet_insights.ipynb
@@ -355,6 +355,9 @@
     "        gmin = weights[0:nrow, 0:ncol].min()\n",
     "        gmax = weights[0:nrow, 0:ncol].max()\n",
     "        \n",
+    "    if isinstance(gmin, torch.Tensor):\n",
+    "        gmin = gmin.item()\n",
+    "        gmax = gmax.item()\n",
     "    return gmin, gmax\n",
     "\n",
     "def plot_param_kernels(model, weights, layout, size_ctrl, binary_mask=False, color_normalization='Model', \n",