Ah, I see here is the reason why above both `dx` and `dk` are calculated even if `dx_is_dk` is `True`...