diff --git a/seq2seq-translation/seq2seq-translation-batched.ipynb b/seq2seq-translation/seq2seq-translation-batched.ipynb
index 465ce6f..19531ea 100644
--- a/seq2seq-translation/seq2seq-translation-batched.ipynb
+++ b/seq2seq-translation/seq2seq-translation-batched.ipynb
@@ -1289,8 +1289,8 @@
    "outputs": [],
    "source": [
     "def evaluate(input_seq, max_length=MAX_LENGTH):\n",
-    "    input_lengths = [len(input_seq)]\n",
     "    input_seqs = [indexes_from_sentence(input_lang, input_seq)]\n",
+    "    input_lengths = [len(input_seqs[0])]\n",
     "    input_batches = Variable(torch.LongTensor(input_seqs), volatile=True).transpose(0, 1)\n",
     "    \n",
     "    if USE_CUDA:\n",