diff --git a/chapters/en/unit3/vision-transformers/vision-transformers-for-image-segmentation.mdx b/chapters/en/unit3/vision-transformers/vision-transformers-for-image-segmentation.mdx
index d807fb5ac..21377a6f9 100644
--- a/chapters/en/unit3/vision-transformers/vision-transformers-for-image-segmentation.mdx
+++ b/chapters/en/unit3/vision-transformers/vision-transformers-for-image-segmentation.mdx
@@ -56,7 +56,7 @@ segmentation = pipeline("image-segmentation", "facebook/maskformer-swin-base-coc
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
-results = segmentation(images=image, subtask="panoptic")
+results = segmentation(inputs=image, subtask="panoptic")
 results
 ```