diff --git a/parcels/collection/collectionsoa.py b/parcels/collection/collectionsoa.py
index 3d24c0867..f739acad8 100644
--- a/parcels/collection/collectionsoa.py
+++ b/parcels/collection/collectionsoa.py
@@ -100,7 +100,7 @@ def __init__(self, pclass, lon, lat, depth, time, lonlatdepth_dtype, pid_orig, p
 
             if mpi_size > 1:
                 if partitions is not False:
-                    if self._pu_indicators is None:
+                    if (self._pu_indicators is None) or (len(self._pu_indicators) != len(lon)):
                         if mpi_rank == 0:
                             coords = np.vstack((lon, lat)).transpose()
                             kmeans = KMeans(n_clusters=mpi_size, random_state=0).fit(coords)