diff --git a/fine-tune-w2v2-bert.md b/fine-tune-w2v2-bert.md
index 018c2edc72..2886af12c3 100644
--- a/fine-tune-w2v2-bert.md
+++ b/fine-tune-w2v2-bert.md
@@ -749,7 +749,7 @@ Finally, we can decode the example from the predicted tokens and compare it to t
 
 ```python
 print(processor.decode(pred_ids))
-print(processor.decode(input_dict["labels"]).lower())
+print(processor.decode(sample["labels"]).lower())
 ```
 
 ```bash