diff --git a/clusterization/kmeans/kmeans.py b/clusterization/kmeans/kmeans.py
index 852844d..7b4ae1c 100644
--- a/clusterization/kmeans/kmeans.py
+++ b/clusterization/kmeans/kmeans.py
@@ -110,10 +110,7 @@ def _assign(self):
             cluster_index = None
             for ci, c in enumerate(self.clusters):
                 distance = p.distance(c.mean)
-                if min_distance is None:
-                    min_distance = distance
-                    cluster_index = ci
-                elif distance < min_distance:
+                if min_distance is None or distance < min_distance:
                     min_distance = distance
                     cluster_index = ci
             tmp_clusters[cluster_index].append(p)