Fix kmeans tests (cf. #121)