diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index 1b113a5..b0dea2b 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.ipynb
@@ -868,7 +868,8 @@
     "    if val_error < minimum_val_error:\n",
     "        minimum_val_error = val_error\n",
     "        best_epoch = epoch\n",
-    "        best_model = clone(sgd_reg)"
+    "        best_model = sgd_reg\n",
+    "        sgd_reg = clone(sgd_reg)"
    ]
   },
   {