diff --git a/tests/01_cluster_splitter.py b/tests/01_cluster_splitter.py
index 08ed30d..e7353b8 100644
--- a/tests/01_cluster_splitter.py
+++ b/tests/01_cluster_splitter.py
@@ -37,7 +37,7 @@ def plot_clusters(cities, clusters):
 
 nb_ville = 100
 max_coords = 1000
-nb_truck = 4
+nb_truck = 5
 
 # Define the coordinates of the cities
 # And set depot at the first city in the middle of the map
diff --git a/tests/clustering.py b/tests/clustering.py
index fb8c959..02ff6df 100644
--- a/tests/clustering.py
+++ b/tests/clustering.py
@@ -13,7 +13,7 @@ def split_tour_across_clusters(cities, nb_truck):
     for i, label in enumerate(kmeans.labels_):
         clusters[label].append(i)
 
-    max_iterations = len(cities)**2
+    max_iterations = len(cities)
     iteration = 0
 
     while True: