A few test tweaks with gradients