Fix setDataPartition function