diff --git a/notebooks/sam_finetuning.ipynb b/notebooks/sam_finetuning.ipynb
index c0b1a62d..d1773169 100644
--- a/notebooks/sam_finetuning.ipynb
+++ b/notebooks/sam_finetuning.ipynb
@@ -219,14 +219,10 @@
     "\n",
     "import torch\n",
     "\n",
-    "import torch_em\n",
     "from torch_em.util.debug import check_loader\n",
     "from torch_em.data import MinInstanceSampler\n",
-    "from torch_em.loss import DiceBasedDistanceLoss\n",
     "from torch_em.util.util import get_random_colors\n",
-    "from torch_em.transform.label import PerObjectDistanceTransform\n",
     "\n",
-    "from micro_sam import util\n",
     "import micro_sam.training as sam_training\n",
     "from micro_sam.sample_data import fetch_tracking_example_data, fetch_tracking_segmentation_data\n",
     "from micro_sam.automatic_segmentation import get_predictor_and_segmenter, automatic_instance_segmentation"