diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index e6bf1d9..12d358f 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.ipynb
@@ -1075,7 +1075,7 @@
     "plt.annotate('Best model',\n",
     "             xy=(best_epoch, best_valid_rmse),\n",
     "             xytext=(best_epoch, best_valid_rmse + 0.5),\n",
-    "             ha=\"center\", fontsize=14,\n",
+    "             ha=\"center\",\n",
     "             arrowprops=dict(facecolor='black', shrink=0.05))\n",
     "plt.plot([0, n_epochs], [best_valid_rmse, best_valid_rmse], \"k:\", linewidth=2)\n",
     "plt.plot(val_errors, \"b-\", linewidth=3, label=\"Validation set\")\n",
@@ -1288,8 +1288,8 @@
     "contour = plt.contour(x0, x1, zz, cmap=plt.cm.brg)\n",
     "plt.clabel(contour, inline=1)\n",
     "plt.plot(left_right, boundary, \"k--\", linewidth=3)\n",
-    "plt.text(3.5, 1.27, \"Not Iris virginica\", color=\"b\", ha=\"center\", fontsize=14)\n",
-    "plt.text(6.5, 2.3, \"Iris virginica\", color=\"g\", ha=\"center\", fontsize=14)\n",
+    "plt.text(3.5, 1.27, \"Not Iris virginica\", color=\"b\", ha=\"center\")\n",
+    "plt.text(6.5, 2.3, \"Iris virginica\", color=\"g\", ha=\"center\")\n",
     "plt.xlabel(\"Petal length\")\n",
     "plt.ylabel(\"Petal width\")\n",
     "plt.axis([2.9, 7, 0.8, 2.7])\n",