Fix TokenPartitionedTopology tests.