diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index b0dea2b..0ac4d1a 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.ipynb
@@ -845,7 +845,7 @@
    },
    "outputs": [],
    "source": [
-    "from sklearn.base import clone\n",
+    "from copy import deepcopy\n",
     "\n",
     "poly_scaler = Pipeline([\n",
     "        (\"poly_features\", PolynomialFeatures(degree=90, include_bias=False)),\n",
@@ -868,8 +868,7 @@
     "    if val_error < minimum_val_error:\n",
     "        minimum_val_error = val_error\n",
     "        best_epoch = epoch\n",
-    "        best_model = sgd_reg\n",
-    "        sgd_reg = clone(sgd_reg)"
+    "        best_model = deepcopy(sgd_reg)"
    ]
   },
   {