diff --git a/examples/konnektor_example.ipynb b/examples/konnektor_example.ipynb
index 70a5ac5..40f1df7 100644
--- a/examples/konnektor_example.ipynb
+++ b/examples/konnektor_example.ipynb
@@ -71,7 +71,7 @@
     }
    ],
    "source": [
-    "from konnektor.visualization import draw_ligand_network\n",
+    "from konnektor.visualization.visualization import draw_ligand_network\n",
     "\n",
     "system= benzenes.get_system()\n",
     "compounds = system.ligand_components\n",
@@ -461,7 +461,7 @@
     "from konnektor.network_connecting_algorithms.bipartite_MST_connect import log\n",
     "log.setLevel(logging.ERROR)\n",
     "from kartograf.atom_align import align_mol_shape\n",
-    "from konnektor.visualization import draw_ligand_network\n",
+    "from konnektor.visualization.visualization import draw_ligand_network\n",
     "from openfe.setup.atom_mapping.lomap_scorers import default_lomap_score\n",
     "\n",
     "tset=[benzenes, hif2a,\n",