diff --git a/site/en/tutorials/images/segmentation.ipynb b/site/en/tutorials/images/segmentation.ipynb
index cd4e583aafc..c2279185b89 100644
--- a/site/en/tutorials/images/segmentation.ipynb
+++ b/site/en/tutorials/images/segmentation.ipynb
@@ -182,7 +182,11 @@
       "source": [
         "def load_image(datapoint):\n",
         "  input_image = tf.image.resize(datapoint['image'], (128, 128))\n",
-        "  input_mask = tf.image.resize(datapoint['segmentation_mask'], (128, 128))\n",
+        "  input_mask = tf.image.resize(\n",
+        "    datapoint['segmentation_mask'],\n",
+        "    (128, 128),\n",
+        "    method = tf.image.ResizeMethod.NEAREST_NEIGHBOR,\n",
+        "  )\n",
         "\n",
         "  input_image, input_mask = normalize(input_image, input_mask)\n",
         "\n",