diff --git a/python/mjspec.ipynb b/python/mjspec.ipynb
index f59cac69..6f9455fd 100644
--- a/python/mjspec.ipynb
+++ b/python/mjspec.ipynb
@@ -8,7 +8,7 @@
    "source": [
     "![MuJoCo banner](https://raw.githubusercontent.com/google-deepmind/mujoco/main/banner.png)\n",
     "\n",
-    "# <h1><center>Model Editing  <a href=\"https://colab.research.google.com/github/google-deepmind/mujoco/blob/main/python/mjSpec.ipynb\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" width=\"140\" align=\"center\"/></a></center></h1>\n",
+    "# <h1><center>Model Editing  <a href=\"https://colab.research.google.com/github/google-deepmind/mujoco/blob/main/python/mjspec.ipynb\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" width=\"140\" align=\"center\"/></a></center></h1>\n",
     "\n",
     "This notebook provides an introductory tutorial for model editing in MuJoCo using the `mjSpec` API. This notebook assumes that the reader is already familiar with MuJoCo basic concepts, as demostrated in the [introductory tutorial](https://github.com/google-deepmind/mujoco?tab=readme-ov-file#getting-started). Documentation for this API can be found in the [Model Editing](https://mujoco.readthedocs.io/en/latest/programming/modeledit.html) chapter in the documentation (C API) and in the [Python chapter](https://mujoco.readthedocs.io/en/latest/python.html#model-editing). Here we use the Python API.\n",
     "\n",
@@ -83,8 +83,8 @@
     "# Check if installation was succesful.\n",
     "try:\n",
     "  print('Checking that the installation succeeded:')\n",
-    "  import mujoco\n",
-    "  mujoco.MjModel.from_xml_string('<mujoco/>')\n",
+    "  import mujoco as mj\n",
+    "  mj.MjModel.from_xml_string('<mujoco/>')\n",
     "except Exception as e:\n",
     "  raise e from RuntimeError(\n",
     "      'Something went wrong during installation. Check the shell output above '\n",
@@ -806,9 +806,6 @@
  ],
  "metadata": {
   "colab": {
-   "collapsed_sections": [
-    "sJFuNetilv4m"
-   ],
    "private_outputs": true
   },
   "kernelspec": {