Should partitionBy before returning