found an error in PartitionedAssign and PartitionedRepl