modify compute method

modify compute method