diff --git a/wario/contact_clustering.ipynb b/wario/contact_clustering.ipynb
index 6a317fbcea34b7d120c9dd31f380b7484e88d2d6..242389acfdd1e3a71510d9b2ecea0c32d26fa573 100644
--- a/wario/contact_clustering.ipynb
+++ b/wario/contact_clustering.ipynb
@@ -252,7 +252,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "get_cluster_files(ensemble_path = ensemble_folder, ensemble_name = ensemble_name, clustering_partition = labels_umap)"
+    "get_cluster_files(ensemble_path = ensemble_folder, ensemble_name = ensemble_name, labels_umap = labels_umap)"
    ]
   },
   {
@@ -278,7 +278,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "representative_ensemble(size = 10, ensemble_path = ensemble_folder, ensemble_name = ensemble_name, clustering_partition = labels_umap)"
+    "representative_ensemble(size = 10, ensemble_path = ensemble_folder, ensemble_name = ensemble_name, labels_umap = labels_umap)"
    ]
   },
   {
@@ -298,7 +298,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "cluster_descriptors(ensemble_path = ensemble_folder, ensemble_name = ensemble_name, clustering_partition = labels_umap)"
+    "cluster_descriptors(ensemble_path = ensemble_folder, ensemble_name = ensemble_name, labels_umap = labels_umap)"
    ]
   }
  ],