fix: grad param added