diff --git a/3d_segmentation/vista3d_spleen_finetune.ipynb b/3d_segmentation/vista3d_spleen_finetune.ipynb
index ad63983aa..4c61fdda5 100644
--- a/3d_segmentation/vista3d_spleen_finetune.ipynb
+++ b/3d_segmentation/vista3d_spleen_finetune.ipynb
@@ -87,7 +87,7 @@
     "    ScaleIntensityRanged,\n",
     "    Spacingd,\n",
     "    Invertd,\n",
-    "    LoadImage\n",
+    "    LoadImage,\n",
     ")\n",
     "from monai.apps import download_url\n",
     "from monai.networks.nets import vista3d132\n",
@@ -578,7 +578,7 @@
     "                    {\"pred\": val_outputs[0], \"label_prompt\": torch.tensor(label_set[1:], device=device)}\n",
     "                )[\"pred\"]\n",
     "                _dice_metric = []\n",
-    "                # Must loop to calculate dice for incontinuous class indexes. 0 is excluded. \n",
+    "                # Must loop to calculate dice for incontinuous class indexes. 0 is excluded.\n",
     "                for i in label_set[1:]:\n",
     "                    _dice_metric.append(\n",
     "                        compute_dice(\n",
@@ -940,7 +940,7 @@
     "i = 0\n",
     "with torch.no_grad():\n",
     "    for test_data in test_org_loader:\n",
-    "        # only plot 2 examples. \n",
+    "        # only plot 2 examples.\n",
     "        if i > 1:\n",
     "            break\n",
     "        i += 1\n",