diff --git a/README.md b/README.md
index f8dbf80..bc18613 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 [![Tests][badge-tests]][link-tests]
 [![Documentation][badge-docs]][link-docs]
 
-[badge-tests]: https://img.shields.io/github/actions/workflow/status/colganwi/pycea/test.yaml?branch=main
+[badge-tests]: https://img.shields.io/github/actions/workflow/status/YosefLab/pycea/test.yaml?branch=main
 [link-tests]: https://github.com/YosefLab/pycea/actions/workflows/test.yml
 [badge-docs]: https://img.shields.io/readthedocs/pycea
 
diff --git a/docs/api.md b/docs/api.md
index 54885f0..b76f6c7 100644
--- a/docs/api.md
+++ b/docs/api.md
@@ -33,6 +33,5 @@
 .. autosummary::
     :toctree: generated
 
-    pl.basic_plot
-    pl.BasicClass
+    pl.branches
 ```