fix gradient issue
update