Fix gradient in example