diff --git a/demo/MMSegmentation_Tutorial.ipynb b/demo/MMSegmentation_Tutorial.ipynb
index 0eb7b5a03062e42e9efcff836543b7fe47cd0212..a75ec5c575290f60992ad637708cbd8aeb1a0326 100644
--- a/demo/MMSegmentation_Tutorial.ipynb
+++ b/demo/MMSegmentation_Tutorial.ipynb
@@ -561,8 +561,7 @@
     "datasets = [build_dataset(cfg.data.train)]\n",
     "\n",
     "# Build the detector\n",
-    "model = build_segmentor(\n",
-    "    cfg.model, train_cfg=cfg.get('train_cfg'), test_cfg=cfg.get('test_cfg'))\n",
+    "model = build_segmentor(cfg.model)\n",
     "# Add an attribute for visualization convenience\n",
     "model.CLASSES = datasets[0].CLASSES\n",
     "\n",