Fix in the nesterov method as well