Small fix to KL divergence prior