fix kl divergence for gaining knowledge of a new staet