diff --git a/scripts/_shape_embed.py b/scripts/_shape_embed.py
index 3ede8763..5e8b11b9 100644
--- a/scripts/_shape_embed.py
+++ b/scripts/_shape_embed.py
@@ -23,7 +23,7 @@
 
 torch.multiprocessing.set_sharing_strategy("file_system")
 
-from bioimage_embed import shapes
+from shape_embed import shapes
 import bioimage_embed
 
 # Note - you must have torchvision installed for this example
diff --git a/scripts/shape_embed.py b/scripts/shape_embed.py
index 73764102..f1064f7c 100644
--- a/scripts/shape_embed.py
+++ b/scripts/shape_embed.py
@@ -33,14 +33,14 @@
 
 torch.multiprocessing.set_sharing_strategy("file_system")
 
-from bioimage_embed import shapes
+from shape_embed import shapes
 import bioimage_embed
 from pytorch_lightning import loggers as pl_loggers
 from torchvision import transforms
 from bioimage_embed.lightning import DataModule
 
 from torchvision import datasets
-from bioimage_embed.shapes.transforms import (
+from shape_embed.shapes.transforms import (
     ImageToCoords,
     CropCentroidPipeline,
     DistogramToCoords,