diff --git a/ch12/ch12_part2.ipynb b/ch12/ch12_part2.ipynb
index 890355504..e512ddbaf 100644
--- a/ch12/ch12_part2.ipynb
+++ b/ch12/ch12_part2.ipynb
@@ -422,7 +422,7 @@
     "fig = plt.figure(figsize=(13, 5))\n",
     "ax = fig.add_subplot(1, 2, 1)\n",
     "plt.plot(X_train_norm.detach().numpy(), y_train.detach().numpy(), 'o', markersize=10)\n",
-    "plt.plot(X_test_norm.detach().numpy(), y_pred.detach().numpy(), '--', lw=3)\n",
+    "plt.plot(X_test_norm.detach().numpy(), y_pred, '--', lw=3)\n",
     "plt.legend(['Training examples', 'Linear reg.'], fontsize=15)\n",
     "ax.set_xlabel('x', size=15)\n",
     "ax.set_ylabel('y', size=15)\n",