diff --git a/examples/unet_segmentation_3d.ipynb b/examples/unet_segmentation_3d.ipynb
index 5a305be150..81ee6ce8b2 100644
--- a/examples/unet_segmentation_3d.ipynb
+++ b/examples/unet_segmentation_3d.ipynb
@@ -36,6 +36,7 @@
     "import nibabel as nib\n",
     "\n",
     "from ignite.engine import Events, create_supervised_trainer\n",
+    "from ignite.handlers import ModelCheckpoint\n",
     "\n",
     "# assumes the framework is found here, change as necessary\n",
     "sys.path.append(\"..\")\n",
@@ -205,6 +206,12 @@
     "\n",
     "trainer = create_supervised_trainer(net, opt, loss_fn, device, False)\n",
     "\n",
+    "checkpoint_handler = ModelCheckpoint('./', 'net', n_saved=10, require_empty=False)\n",
+    "trainer.add_event_handler(\n",
+    "    event_name=Events.EPOCH_COMPLETED,\n",
+    "    handler=checkpoint_handler,\n",
+    "    to_save={'net': net}\n",
+    ")\n",
     "\n",
     "@trainer.on(Events.EPOCH_COMPLETED)\n",
     "def log_training_loss(engine):\n",