Fix kmeans return value?