diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 87776f6f4..067f19695 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -1,6 +1,7 @@
 # This workflow builds the torchrec docs and deploys them to gh-pages.
 name: Generate documentation
 on:
+  pull_request:
   push:
     branches:
       - main
diff --git a/docs/source/torchrec.modules.rst b/docs/source/torchrec.modules.rst
index 95dcf35db..e2c9ee771 100644
--- a/docs/source/torchrec.modules.rst
+++ b/docs/source/torchrec.modules.rst
@@ -76,10 +76,18 @@ torchrec.modules.utils
    :undoc-members:
    :show-inheritance:
 
-Module contents
+torchrec.modules.mc_modules
 ---------------
 
-.. automodule:: torchrec.modules
+.. automodule:: torchrec.modules.mc_modules
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+torchrec.modules.mc_embedding_modules
+---------------
+
+.. automodule:: torchrec.modules.mc_embedding_modules
    :members:
    :undoc-members:
    :show-inheritance: