diff --git a/exp/hinge_characterization_3dp_pla_shims/stiffness_identification.ipynb b/exp/hinge_characterization_3dp_pla_shims/stiffness_identification.ipynb
index db8e963..3c218ac 100644
--- a/exp/hinge_characterization_3dp_pla_shims/stiffness_identification.ipynb
+++ b/exp/hinge_characterization_3dp_pla_shims/stiffness_identification.ipynb
@@ -52,7 +52,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -268,7 +268,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -281,27 +281,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "(<Figure size 900x300 with 3 Axes>,\n",
-       " array([<Axes: title={'center': 'tension'}, xlabel='Displacement [mm]', ylabel='Force [N]'>,\n",
-       "        <Axes: title={'center': 'compression'}, xlabel='Displacement [mm]', ylabel='Force [N]'>,\n",
-       "        <Axes: title={'center': 'shear'}, xlabel='Displacement [mm]', ylabel='Force [N]'>],\n",
-       "       dtype=object))"
-      ]
-     },
-     "execution_count": 24,
-     "metadata": {},
-     "output_type": "execute_result"
-    },
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "fc614706ec084c968a935cb678fc68ea",
+       "model_id": "e6773f40ec6a4b4cba434102ac09cb80",
        "version_major": 2,
        "version_minor": 0
       },
@@ -325,12 +311,15 @@
     }
    ],
    "source": [
-    "plot_force_displacement_fitting(\n",
+    "fig, axes = plot_force_displacement_fitting(\n",
     "    exp_data=optimization.target_responses,\n",
     "    sim_data=optimization.fitted_responses,\n",
     "    figsize=(9, 3),\n",
     "    lw=2.,\n",
-    ")"
+    ")\n",
+    "out_path = out_folder / \"force_displacement_fitting.png\"\n",
+    "out_path.parent.mkdir(parents=True, exist_ok=True)\n",
+    "fig.savefig(out_path, dpi=300)"
    ]
   },
   {