fix k_means tests