modify return value of partial_argsort