diff --git a/docs/report/000.ipynb b/docs/report/000.ipynb
index 9495740c..8979a3ac 100644
--- a/docs/report/000.ipynb
+++ b/docs/report/000.ipynb
@@ -16,6 +16,17 @@
     "<!-- cspell:disable -->"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    ":::{note}\n",
+    "\n",
+    "This report has been implemented in [ComPWA/tensorwaves#284](https://github.com/ComPWA/tensorwaves/pull/284).\n",
+    "\n",
+    ":::"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
diff --git a/docs/report/001.ipynb b/docs/report/001.ipynb
index 862ac83c..19228112 100644
--- a/docs/report/001.ipynb
+++ b/docs/report/001.ipynb
@@ -7,6 +7,17 @@
     "# [TR-001] Custom lambdification"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    ":::{note}\n",
+    "\n",
+    "This report has been implemented in [ComPWA/ampform#72](https://github.com/ComPWA/ampform/pull/72) and [ComPWA/ampform#284](https://github.com/ComPWA/tensorwaves/pull/284).\n",
+    "\n",
+    ":::"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -414,7 +425,9 @@
     "    _module = \"jax\"\n",
     "\n",
     "    def _print_ComplexSqrt(self, expr: sp.Expr) -> str:\n",
-    "        return \"select([less(x, 0), True], [1j * sqrt(-x), sqrt(x)], default=nan,)\""
+    "        arg = expr.args[0]\n",
+    "        x = self._print(arg)\n",
+    "        return f\"select([less({x}, 0), True], [1j * sqrt(-{x}), sqrt({x})], default=nan,)\""
    ]
   },
   {
diff --git a/docs/report/002.ipynb b/docs/report/002.ipynb
index bc4464b3..e78723fe 100644
--- a/docs/report/002.ipynb
+++ b/docs/report/002.ipynb
@@ -14,6 +14,17 @@
     "<!-- cspell:disable -->"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    ":::{note}\n",
+    "\n",
+    "This report has been implemented in [ComPWA/tensorwaves#281](https://github.com/ComPWA/tensorwaves/pull/281).\n",
+    "\n",
+    ":::"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,