diff --git a/image_segmentation_conditional_random_fields.ipynb b/image_segmentation_conditional_random_fields.ipynb
index bd7cfbc..d174ab1 100644
--- a/image_segmentation_conditional_random_fields.ipynb
+++ b/image_segmentation_conditional_random_fields.ipynb
@@ -764,7 +764,7 @@
     "\n",
     "softmax = final_probabilities.squeeze()\n",
     "\n",
-    "softmax = processed_probabilities.transpose((2, 0, 1))\n",
+    "processed_probabilities = softmax.transpose((2, 0, 1))\n",
     "\n",
     "# The input should be the negative of the logarithm of probability values\n",
     "# Look up the definition of the softmax_to_unary for more information\n",