diff --git a/docs/api/mass.rst b/docs/api/mass.rst
index 368f617cb..cccfab6ce 100644
--- a/docs/api/mass.rst
+++ b/docs/api/mass.rst
@@ -24,8 +24,8 @@ Total [ag.mp]
    IsothermalSph
    dPIEMass
    dPIEMassSph
-   dPIEMassLenstool
-   dPIEMassLenstoolSph
+   dPIEMassB0
+   dPIEMassB0Sph
    PIEMass
    dPIEPotential
    dPIEPotentialSph
diff --git a/docs/api/potential_correction.rst b/docs/api/potential_correction.rst
index 0f074044d..a2326beb4 100644
--- a/docs/api/potential_correction.rst
+++ b/docs/api/potential_correction.rst
@@ -13,7 +13,7 @@ functionality in your research, please cite Cao et al. 2025; citation
 materials are provided at
 https://github.com/caoxiaoyue/potential_correction_paper.
 
-.. currentmodule:: autolens.pc
+.. currentmodule:: autolens.potential_correction
 
 .. autosummary::
    :toctree: _autosummary