:bug: BugFix in gradient stepping