diff --git a/src/cellmap_models/pytorch/cellpose/README.md b/src/cellmap_models/pytorch/cellpose/README.md
index 4b2db52..37ae3ef 100644
--- a/src/cellmap_models/pytorch/cellpose/README.md
+++ b/src/cellmap_models/pytorch/cellpose/README.md
@@ -62,7 +62,7 @@ Once you have chosen a model based on the descriptions above, you can download i
 If you would like to load a model for your own use, you can do the following:
 
 ```python
-from cellmap_models.cellpose import load_model
+from cellmap_models.pytorch.cellpose import load_model
 model = load_model('<model_name>')
 ```
 
diff --git a/src/cellmap_models/pytorch/cosem/README.md b/src/cellmap_models/pytorch/cosem/README.md
index 1629ccb..f25a786 100755
--- a/src/cellmap_models/pytorch/cosem/README.md
+++ b/src/cellmap_models/pytorch/cosem/README.md
@@ -9,7 +9,7 @@ The networks have been converted to PyTorch from their original Tensorflow versi
 You can load a PyTorch model using the following code:
 
 ```python
-import cellmap_models.cosem as cosem_models
+import cellmap_models.pytorch.cosem as cosem_models
 model = cosem_models.load_model('setup04/1820500')
 # The model is now ready to use
 ```
@@ -23,7 +23,7 @@ cosem_models.models_list
 Each model has a separate backbone and single layer prediction head. The `backbone` and `head` objects are both PyTorch modules and can be used as such. You can access the separate components of the model using the following code:
 
 ```python
-import cellmap_models.cosem as cosem_models
+import cellmap_models.pytorch.cosem as cosem_models
 model = cosem_models.load_model('setup04/1820500')
 backbone = model.backbone
 head = model.prediction_head