diff --git a/docs/api.md b/docs/api.md
index 515c44d..7aaa36d 100644
--- a/docs/api.md
+++ b/docs/api.md
@@ -12,7 +12,7 @@ import scanvi_explainer
 ## SCANVIDeep
 
 ```{eval-rst}
-.. currentmodule:: scanvi_explainer
+.. currentmodule:: scanvi_explainer.scanvi_deep
 ```
 
 ```{eval-rst}
@@ -20,13 +20,13 @@ import scanvi_explainer
     :toctree: generated/
     :nosignatures:
 
-    scanvi_deep.SCANVIDeep
+    SCANVIDeep
 ```
 
 ## Utilities
 
 ```{eval-rst}
-.. currentmodule:: scanvi_explainer
+.. currentmodule:: scanvi_explainer.utils
 ```
 
 ```{eval-rst}
@@ -34,13 +34,13 @@ import scanvi_explainer
     :toctree: generated/
     :nosignatures:
 
-    utils.train_test_group_split
+    train_test_group_split
 ```
 
 ## Plots
 
 ```{eval-rst}
-.. currentmodule:: scanvi_explainer
+.. currentmodule:: scanvi_explainer.plots
 ```
 
 ```{eval-rst}
@@ -48,5 +48,5 @@ import scanvi_explainer
     :toctree: generated/
     :nosignatures:
 
-    plots.feature_plot
+    feature_plot
 ```
\ No newline at end of file