fix kmeans demo