diff --git a/README.md b/README.md
index bf5453853197..eb337d8253f4 100644
--- a/README.md
+++ b/README.md
@@ -142,7 +142,7 @@ predictions = model(tokens_tensor, segments_tensors)
 
 # confirm we were able to predict 'henson'
 predicted_index = torch.argmax(predictions[0, masked_index]).item()
-predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])
+predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0]
 assert predicted_token == 'henson'
 ```