diff --git a/notebooks/npt_simulation.ipynb b/notebooks/npt_simulation.ipynb
index c7e5dc36..27aa897c 100644
--- a/notebooks/npt_simulation.ipynb
+++ b/notebooks/npt_simulation.ipynb
@@ -22,7 +22,7 @@
         "colab_type": "text"
       },
       "source": [
-        "<a href=\"https://colab.research.google.com/github/google/jax-md/blob/npt_fix/notebooks/npt_simulation.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+        "<a href=\"https://colab.sandbox.google.com/github/google/jax-md/blob/main/notebooks/npt_simulation.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
     {
@@ -900,4 +900,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}