Fix KMeans method call