diff --git a/char-rnn-generation/char-rnn-generation.ipynb b/char-rnn-generation/char-rnn-generation.ipynb
index dbeeebf..dbdfabb 100644
--- a/char-rnn-generation/char-rnn-generation.ipynb
+++ b/char-rnn-generation/char-rnn-generation.ipynb
@@ -341,12 +341,12 @@
     "\n",
     "    for c in range(chunk_len):\n",
     "        output, hidden = decoder(inp[c], hidden)\n",
-    "        loss += criterion(output, target[c])\n",
+    "        loss += criterion(output, target[c].view(1))\n",
     "\n",
     "    loss.backward()\n",
     "    decoder_optimizer.step()\n",
     "\n",
-    "    return loss.data[0] / chunk_len"
+    "    return loss.data.item() / chunk_len"
    ]
   },
   {