diff --git a/examples/nvt_jaxmd.ipynb b/examples/nvt_jaxmd.ipynb
index 461058a..fede587 100644
--- a/examples/nvt_jaxmd.ipynb
+++ b/examples/nvt_jaxmd.ipynb
@@ -356,7 +356,7 @@
     "\n",
     "# Set the temprature at initialization\n",
     "if T_init is None:\n",
-    "    T_init = float(T_nvt / nvt_cycles) \n",
+    "    T_init = T_nvt \n",
     "else:\n",
     "    T_init = T_init\n",
     "\n",