diff --git a/README.md b/README.md
index 4bd7003..c706bff 100644
--- a/README.md
+++ b/README.md
@@ -46,7 +46,7 @@ Here is how to use this model:
 >>> masks_queries_logits = outputs.masks_queries_logits
 
 >>> # you can pass them to feature_extractor for postprocessing
->>> output = feature_extractor.post_process_semantic_segmentation(outputs)
+>>> predicted_semantic_map = feature_extractor.post_process_semantic_segmentation(outputs)[0]
 ```
 
 For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/maskformer).
\ No newline at end of file