Update gradient again to be correct.