diff --git a/examples/medical_image_segmentation.ipynb b/examples/medical_image_segmentation.ipynb
index 820979f523..25b3bb15cd 100644
--- a/examples/medical_image_segmentation.ipynb
+++ b/examples/medical_image_segmentation.ipynb
@@ -578,8 +578,8 @@
     "        return self.criterion(outputs, targets)\n",
     "\n",
     "    def get_metrics(self, is_train: bool = False):\n",
-    "        if self.is_train:\n",
-    "            return {'BCEWithLogitsLoss', self.train_loss}\n",
+    "        if is_train:\n",
+    "            return {'BCEWithLogitsLoss': self.train_loss}\n",
     "        else:\n",
     "            return {'BCEWithLogitsLoss': self.val_loss, 'Dice': self.dice}"
    ]