diff --git a/notebooks/002_Wakefield_simulation.ipynb b/notebooks/002_Wakefield_simulation.ipynb
index 523087e..0e2a49c 100644
--- a/notebooks/002_Wakefield_simulation.ipynb
+++ b/notebooks/002_Wakefield_simulation.ipynb
@@ -300,6 +300,22 @@
     "fig.savefig('002_results/benchmark.png')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Wakefield results from previous runs can be easily loaded (longitudinal and transverse at once) using:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "wake.load_results(folder='002_results/')"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -415,29 +431,6 @@
     "               plot3d=True, plot_every=100, \n",
     "               **plotkw3D)"
    ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "#### Load wake results"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "Wakefield results from previous runs can be easily loaded (longitudinal and transverse at once) using:"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "wake.load_results(folder='002_results/')"
-   ]
   }
  ],
  "metadata": {