diff --git a/README.md b/README.md
index 3888c1a..aef4817 100644
--- a/README.md
+++ b/README.md
@@ -99,7 +99,7 @@ inputs = tokenizer('''def print_prime(n):
    """''', return_tensors="pt", return_attention_mask=False)
 
 outputs = model.generate(**inputs, max_length=200)
-text = tokenizer.batch_decode(outputs[0])
+text = tokenizer.batch_decode(outputs)[0]
 print(text)
 ```