diff --git a/docs/notebooks/ceviche_challenges.ipynb b/docs/notebooks/ceviche_challenges.ipynb
index 41f8ece..a0c2005 100644
--- a/docs/notebooks/ceviche_challenges.ipynb
+++ b/docs/notebooks/ceviche_challenges.ipynb
@@ -25,12 +25,7 @@
    },
    "outputs": [],
    "source": [
-    "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",
+    "from invrs_gym import challenges\n",
     "\n",
     "docstring = challenges.ceviche_beam_splitter().eval_metric.__doc__\n",
     "print(\"\\n\".join([s[8:] for s in docstring.split(\"Args\")[0].split(\"\\n\")[2:-2]]))"