Fix incorrect arguments to kmeans