diff --git a/docs/source/api.rst b/docs/source/api.rst
index 17dffc345873ad85cf3306bde8b41770965c1bef..05bd3573946fc3f5ea88539e201aa520bd33faf2 100644
--- a/docs/source/api.rst
+++ b/docs/source/api.rst
@@ -4,7 +4,7 @@ tree_maker API documentation
 
 NodeJob.py
 ----------
-.. autoclass:: tree_maker.NodeJob.NodeJob
+.. automodule:: tree_maker.NodeJob
     :members:
 
 tag.py
@@ -15,5 +15,5 @@ tag.py
 
 repository.py
 ------
-.. automodule:: tree_maker.tag
+.. automodule:: tree_maker.repository
     :members: