fixed up the kmeans test code