fix some gradients