diff --git a/05_linear_regression.py b/05_linear_regression.py
index 0afd430..542489e 100644
--- a/05_linear_regression.py
+++ b/05_linear_regression.py
@@ -41,7 +41,7 @@ def forward(self, x):
 
     # Compute and print loss
     loss = criterion(y_pred, y_data)
-    print(epoch, loss.data[0])
+    print(epoch, loss.data)
 
     # Zero gradients, perform a backward pass, and update the weights.
     optimizer.zero_grad()