diff --git a/modules/transforms_demo_2d.ipynb b/modules/transforms_demo_2d.ipynb
index 138e1b929d..af17b76b32 100644
--- a/modules/transforms_demo_2d.ipynb
+++ b/modules/transforms_demo_2d.ipynb
@@ -286,8 +286,8 @@
     "    device=torch.device(\"cuda:0\"),\n",
     ")\n",
     "# convert both image and segmentation using different interpolation mode\n",
-    "new_img = affine(im_data, (300, 400), mode=\"bilinear\")\n",
-    "new_seg = affine(seg_data, (300, 400), mode=\"nearest\")\n",
+    "new_img, _ = affine(im_data, (300, 400), mode=\"bilinear\")\n",
+    "new_seg, _ = affine(seg_data, (300, 400), mode=\"nearest\")\n",
     "print(new_img.shape, new_seg.shape)"
    ]
   },
@@ -427,7 +427,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.7.10"
   }
  },
  "nbformat": 4,