Improve kld test