Change in partitioning api