kmeans: small fix