Fixed gradient issue.