Update test-gradient-simple.jl