diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index 223fb48..5eda80a 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.ipynb
@@ -640,7 +640,7 @@
     "def plot_learning_curves(model, X, y):\n",
     "    X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=10)\n",
     "    train_errors, val_errors = [], []\n",
-    "    for m in range(1, len(X_train)):\n",
+    "    for m in range(1, len(X_train) + 1):\n",
     "        model.fit(X_train[:m], y_train[:m])\n",
     "        y_train_predict = model.predict(X_train[:m])\n",
     "        y_val_predict = model.predict(X_val)\n",