diff --git a/ch13/ch13_part3_lightning.ipynb b/ch13/ch13_part3_lightning.ipynb
index c05fc774c..1e5379758 100644
--- a/ch13/ch13_part3_lightning.ipynb
+++ b/ch13/ch13_part3_lightning.ipynb
@@ -201,7 +201,7 @@
     "        self.log(\"train_loss\", loss, prog_bar=True)\n",
     "        return loss\n",
     "\n",
-    "    def training_epoch_end(self, outs):\n",
+    "    def on_train_epoch_end(self):\n",
     "        self.log(\"train_acc\", self.train_acc.compute())\n",
     "        self.train_acc.reset()\n",
     "    \n",
@@ -214,8 +214,8 @@
     "        self.log(\"valid_loss\", loss, prog_bar=True)\n",
     "        return loss\n",
     "    \n",
-    "    def validation_epoch_end(self, outs):\n",
-    "        self.log(\"valid_acc\", self.valid_acc.compute(), prog_bar=True)\n",
+    "    def on_validation_epoch_end(self):\n",
+    "        self.log(\"valid_acc\", self.valid_acc.compute())\n",
     "        self.valid_acc.reset()\n",
     "\n",
     "    def test_step(self, batch, batch_idx):\n",