Minor change in autodiff