put gradient back in