Fix kmeans