Fix bug in k-means example.