Bug fix Kmeans