diff --git a/docs/notebooks/ceviche_challenges.ipynb b/docs/notebooks/ceviche_challenges.ipynb
index d895d1e..41f8ece 100644
--- a/docs/notebooks/ceviche_challenges.ipynb
+++ b/docs/notebooks/ceviche_challenges.ipynb
@@ -25,10 +25,15 @@
    },
    "outputs": [],
    "source": [
-    "from invrs_gym import challenges\n",
+    "import os\n",
+    "import contextlib\n",
+    "\n",
+    "with open(os.devnull, \"w\") as f, contextlib.redirect_stdout(f):\n",
+    "    # Avoid printing by `refractiveindex` module on first run.\n",
+    "    from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.ceviche_beam_splitter().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {
diff --git a/docs/notebooks/diffractive_splitter.ipynb b/docs/notebooks/diffractive_splitter.ipynb
index 2d353d7..8c83b4b 100644
--- a/docs/notebooks/diffractive_splitter.ipynb
+++ b/docs/notebooks/diffractive_splitter.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.diffractive_splitter().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {
diff --git a/docs/notebooks/meta_atom_library.ipynb b/docs/notebooks/meta_atom_library.ipynb
index 8f643b6..bf8325c 100644
--- a/docs/notebooks/meta_atom_library.ipynb
+++ b/docs/notebooks/meta_atom_library.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.meta_atom_library().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {
diff --git a/docs/notebooks/metagrating.ipynb b/docs/notebooks/metagrating.ipynb
index 2bcc103..5d1f08e 100644
--- a/docs/notebooks/metagrating.ipynb
+++ b/docs/notebooks/metagrating.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.metagrating().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {
diff --git a/docs/notebooks/metalens.ipynb b/docs/notebooks/metalens.ipynb
index 568542b..fa9a7e2 100644
--- a/docs/notebooks/metalens.ipynb
+++ b/docs/notebooks/metalens.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.metalens().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {
diff --git a/docs/notebooks/photon_extractor.ipynb b/docs/notebooks/photon_extractor.ipynb
index b741148..c9c4b7b 100644
--- a/docs/notebooks/photon_extractor.ipynb
+++ b/docs/notebooks/photon_extractor.ipynb
@@ -28,7 +28,7 @@
     "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.photon_extractor().eval_metric.__doc__\n",
-    "print(\"\\n\".join(docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]))"
+    "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"
    ]
   },
   {