Fix the steepest descent solver