Fix to KMeans