Fix gradient (again)