diff --git a/.github/workflows/uml.yml b/.github/workflows/uml.yml
index e241c21f5..23ee9a9f8 100644
--- a/.github/workflows/uml.yml
+++ b/.github/workflows/uml.yml
@@ -1,17 +1,15 @@
 name: Update the UML Diagrams
 
 on:
-  pull_request:
-    branches: [main]
-    paths:
-      - "pymc_marketing/**"
   push:
+    types: [closed]
     branches: [main]
     paths:
       - "pymc_marketing/**"
 
 jobs:
   build:
+    if: github.event.pull_request.head.repo.owner.login == 'pymc-labs'
     runs-on: ubuntu-latest
     permissions: write-all
     steps:
diff --git a/docs/source/uml/classes_clv.png b/docs/source/uml/classes_clv.png
index 38a15be5c..f3a502998 100644
Binary files a/docs/source/uml/classes_clv.png and b/docs/source/uml/classes_clv.png differ
diff --git a/docs/source/uml/classes_mmm.png b/docs/source/uml/classes_mmm.png
index dee22ebae..113f5b371 100644
Binary files a/docs/source/uml/classes_mmm.png and b/docs/source/uml/classes_mmm.png differ
diff --git a/docs/source/uml/packages_clv.png b/docs/source/uml/packages_clv.png
index 714046122..0af9bfb17 100644
Binary files a/docs/source/uml/packages_clv.png and b/docs/source/uml/packages_clv.png differ
diff --git a/docs/source/uml/packages_mmm.png b/docs/source/uml/packages_mmm.png
index 6b1cb0627..ab0ad869a 100644
Binary files a/docs/source/uml/packages_mmm.png and b/docs/source/uml/packages_mmm.png differ
diff --git a/pymc_marketing/prior.py b/pymc_marketing/prior.py
index b2ad01068..1ed68bb8f 100644
--- a/pymc_marketing/prior.py
+++ b/pymc_marketing/prior.py
@@ -66,7 +66,7 @@
     )
 
 Create a transformed hierarchical normal prior by using the `transform`
-parameter.
+parameter. Here the "sigmoid" transformation comes from `pm.math`.
 
 .. code-block:: python
 
@@ -242,7 +242,8 @@ class Prior:
         Only allowed for Normal distribution.
     transform : str, optional
         The name of the transform to apply to the variable after it is
-        created, by default None or no transform.
+        created, by default None or no transform. The transformation must
+        be available in either `pytensor.tensor` or `pymc.math`.
 
     """