diff --git a/volia/clustering_modules/kmeans_multidistance.py b/volia/clustering_modules/kmeans_multidistance.py
index 2d2971d..5ce8e42 100644
--- a/volia/clustering_modules/kmeans_multidistance.py
+++ b/volia/clustering_modules/kmeans_multidistance.py
@@ -66,7 +66,7 @@ class kmeansMultidistance():
         # Learning k-means model
         results = []
         for i in range(ninit):
-            model = KMeans(k=5,
+            model = KMeans(k=k,
                            maxiter=maxiter,
                            distance=self.distance,
                            record_heterogeneity=[],