diff --git a/moderne_visualizations_misc/call_graph_uml.ipynb b/moderne_visualizations_misc/call_graph_uml.ipynb
index 0cf4496..90c20f6 100644
--- a/moderne_visualizations_misc/call_graph_uml.ipynb
+++ b/moderne_visualizations_misc/call_graph_uml.ipynb
@@ -108,7 +108,8 @@
     "        from_method = make_method(from_name)\n",
     "        to_method = make_method(to_name)\n",
     "        plantuml_code.append(\n",
-    "            f'\"{from_class}::{from_method}\" --> \"{to_class}::{to_method}\"')\n",
+    "            f'\"{from_class}::{from_method}\" --> \"{to_class}::{to_method}\"'\n",
+    "        )\n",
     "\n",
     "    if is_valid_class(to_class) and is_valid_method(to_name):\n",
     "        add_method_to_class(to_class, to_name)\n",
@@ -117,7 +118,8 @@
     "        from_method = make_method(from_name)\n",
     "        to_method = make_method(to_name)\n",
     "        plantuml_code.append(\n",
-    "            f'\"{to_class}::{to_method}\" <-- \"{from_class}::{from_method}\"')\n"
+    "            f'\"{to_class}::{to_method}\" <-- \"{from_class}::{from_method}\"'\n",
+    "        )"
    ]
   },
   {
@@ -141,7 +143,7 @@
     "\n",
     "plantuml_code_string = \"\\n\".join(plantuml_code)\n",
     "\n",
-    "# print(plantuml_code_string)\n"
+    "# print(plantuml_code_string)"
    ]
   },
   {