Fixing grad predict again