Change sort and argsort imports