diff --git a/bin/cluster_kmeans.py b/bin/cluster_kmeans.py
index 230757b..68b110a 100644
--- a/bin/cluster_kmeans.py
+++ b/bin/cluster_kmeans.py
@@ -42,6 +42,7 @@ X = np.asarray([feat_ind[x[0][0]][x[0][3]][1] for x in lst])
 Ks = range(KMIN, KMAX+1)
 for k in Ks:
     kmeans = KMeans(n_clusters=k, n_init=10, random_state=0).fit(X)
+    preds = kmeans.predict(X)
     if ALLINDIR is False:
         subdir = str(k)
         dirname=path.join(OUTDIR, subdir)