fix kl_simple test to account for clamping