diff --git a/lab-11_5_seq2seq.ipynb b/lab-11_5_seq2seq.ipynb
index bd2b438..17c6238 100644
--- a/lab-11_5_seq2seq.ipynb
+++ b/lab-11_5_seq2seq.ipynb
@@ -253,7 +253,7 @@
     "        for ei in range(source_length):\n",
     "            _, encoder_hidden = encoder(source_tensor[ei], encoder_hidden)\n",
     "\n",
-    "        decoder_input = torch.Tensor([[SOS_token]], device=device).long()\n",
+    "        decoder_input = torch.Tensor([[SOS_token]]).long().to(device)\n",
     "        decoder_hidden = encoder_hidden\n",
     "        decoded_words = []\n",
     "\n",