diff --git a/docs/tutorials/pytorch_lightning/distilbert-corn-tripadvisor.ipynb b/docs/tutorials/pytorch_lightning/distilbert-corn-tripadvisor.ipynb
index 3f458f3..b4be7c7 100644
--- a/docs/tutorials/pytorch_lightning/distilbert-corn-tripadvisor.ipynb
+++ b/docs/tutorials/pytorch_lightning/distilbert-corn-tripadvisor.ipynb
@@ -799,7 +799,7 @@
     "        self.log(\"train_loss\", loss)\n",
     "\n",
     "        predicted_labels = corn_label_from_logits(outputs[\"logits\"])\n",
-    "        self.test_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
+    "        self.train_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
     "        self.log(\"train_mae\", self.train_mae, prog_bar=True)\n",
     "        \n",
     "        return loss  # this is passed to the optimizer for training\n",
@@ -1214,7 +1214,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.10"
+   "version": "3.10.6"
   }
  },
  "nbformat": 4,
diff --git a/docs/tutorials/pytorch_lightning/distilbert-crossentropy-tripadvisor.ipynb b/docs/tutorials/pytorch_lightning/distilbert-crossentropy-tripadvisor.ipynb
index cae5f22..5562fba 100644
--- a/docs/tutorials/pytorch_lightning/distilbert-crossentropy-tripadvisor.ipynb
+++ b/docs/tutorials/pytorch_lightning/distilbert-crossentropy-tripadvisor.ipynb
@@ -796,7 +796,7 @@
     "        self.log(\"train_loss\", outputs[\"loss\"])\n",
     "\n",
     "        predicted_labels = torch.argmax(outputs[\"logits\"], 1)\n",
-    "        self.test_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
+    "        self.train_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
     "        self.log(\"train_mae\", self.train_mae, prog_bar=True)\n",
     "        \n",
     "        return outputs[\"loss\"]  # this is passed to the optimizer for training\n",
@@ -1209,7 +1209,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.10"
+   "version": "3.10.6"
   }
  },
  "nbformat": 4,