Fix parameters passing in partitioning steps