diff --git a/docs/notebooks/parse_arxiv_papers.ipynb b/docs/notebooks/parse_arxiv_papers.ipynb
index a299f7d90670..3e1f8e1279bf 100644
--- a/docs/notebooks/parse_arxiv_papers.ipynb
+++ b/docs/notebooks/parse_arxiv_papers.ipynb
@@ -342,9 +342,7 @@
     "                or gt_method[\"full_name\"].lower() in predicted_method.lower()\n",
     "            ):\n",
     "                num_correct_methods += 1\n",
-    "    return {\n",
-    "        \"method_prediction_accuracy\": num_correct_methods / len(predicted_methods)\n",
-    "    }"
+    "    return {\"method_prediction_accuracy\": num_correct_methods / len(predicted_methods)}"
    ]
   },
   {