Trying to replace TotalOrderPartitioner for the built-in one