diff --git a/src/HHbbVV/postprocessing/InferenceAnalysis.ipynb b/src/HHbbVV/postprocessing/InferenceAnalysis.ipynb
index 319a1d6d..16d7069a 100644
--- a/src/HHbbVV/postprocessing/InferenceAnalysis.ipynb
+++ b/src/HHbbVV/postprocessing/InferenceAnalysis.ipynb
@@ -108,9 +108,9 @@
     "]\n",
     "\n",
     "for mX, mY in res_mps:\n",
-    "    res_samples[\n",
-    "        f\"X[{mX}]->H(bb)Y[{mY}](WW)\"\n",
-    "    ] = f\"NMSSM_XToYH_MX{mX}_MY{mY}_HTo2bYTo2W_hadronicDecay\"\n",
+    "    res_samples[f\"X[{mX}]->H(bb)Y[{mY}](WW)\"] = (\n",
+    "        f\"NMSSM_XToYH_MX{mX}_MY{mY}_HTo2bYTo2W_hadronicDecay\"\n",
+    "    )\n",
     "\n",
     "res_sig_keys = list(res_samples.keys())\n",
     "nonres_sig_keys = [\"HHbbVV\"]"
@@ -310,6 +310,7 @@
     "    sample2...\n",
     "}\n",
     "\"\"\"\n",
+    "\n",
     "pt_key = \"Pt\"\n",
     "msd_key = \"Msd\"\n",
     "var_prefix = \"ak8FatJet\"\n",