diff --git a/doc/source/apisrc/pdfflow.rst b/doc/source/apisrc/pdfflow.rst
index a4ba030..48dcbe6 100644
--- a/doc/source/apisrc/pdfflow.rst
+++ b/doc/source/apisrc/pdfflow.rst
@@ -7,7 +7,7 @@ Submodules
 pdfflow.pdfflow module
 ------------------------
 
-.. automodule:: pdfflow.pdfflow
+.. automodule:: pdfflow.pflow
    :members:
    :undoc-members:
    :show-inheritance:
diff --git a/doc/source/conf.py b/doc/source/conf.py
index 736dde1..e13052b 100644
--- a/doc/source/conf.py
+++ b/doc/source/conf.py
@@ -28,7 +28,7 @@
 release = pdfflow.__version__
 
 # The full version, including alpha/beta/rc tags
-autodoc_mock_imports = ['tensorflow']
+autodoc_mock_imports = ['tensorflow', 'tensorflow-probability']
 
 
 # -- General configuration ---------------------------------------------------