fix K for kmeans