diff --git a/char-rnn-classification/train.py b/char-rnn-classification/train.py
index 5831980..9ad2960 100644
--- a/char-rnn-classification/train.py
+++ b/char-rnn-classification/train.py
@@ -42,7 +42,7 @@ def train(category_tensor, line_tensor):
 
     optimizer.step()
 
-    return output, loss.data[0]
+    return output, loss.item()
 
 # Keep track of losses for plotting
 current_loss = 0