diff --git a/docs/notebooks/bold_realignment.ipynb b/docs/notebooks/bold_realignment.ipynb
index 3f433bb89..888541f98 100644
--- a/docs/notebooks/bold_realignment.ipynb
+++ b/docs/notebooks/bold_realignment.ipynb
@@ -164,21 +164,15 @@
     "afni_realigned"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from matplotlib import pyplot as plt"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 43,
    "metadata": {},
    "outputs": [],
    "source": [
+    "import pandas as pd\n",
+    "from matplotlib import pyplot as plt\n",
+    "\n",
     "from nifreeze.viz.motion_viz import plot_framewise_displacement\n",
     "\n",
     "\n",
@@ -264,10 +258,9 @@
     "    )\n",
     "\n",
     "    # Plot the framewise displacement on the first axis\n",
+    "    fd = pd.DataFrame({\"afni\": afni_fd, \"nifreeze\": nifreeze_fd})\n",
     "    fd_axis = axes[0]\n",
-    "    _ = plot_framewise_displacement(\n",
-    "        afni_fd, nifreeze_fd, \"AFNI 3dVolreg FD\", \"nifreeze FD\", ax=fd_axis\n",
-    "    )\n",
+    "    _ = plot_framewise_displacement(fd, [\"AFNI 3dVolreg FD\", \"nifreeze FD\"], ax=fd_axis)\n",
     "\n",
     "    # Set labels for profile plots if not provided\n",
     "    if labels is None or isinstance(labels, str):\n",