Simplify distributor unit tests.

1 files changed, 2 insertions, 3 deletions

diff --git a/storage/src/tests/distributor/twophaseupdateoperationtest.cpp b/storage/src/tests/distributor/twophaseupdateoperationtest.cpp
index 16d317551e0..ed961b29b1c 100644
--- a/storage/src/tests/distributor/twophaseupdateoperationtest.cpp
+++ b/storage/src/tests/distributor/twophaseupdateoperationtest.cpp
@@ -579,7 +579,7 @@ TwoPhaseUpdateOperationTest::testFastPathInconsistentTimestampsPutNotStarted()
             sender.getLastCommand(true));
     checkMessageSettingsPropagatedTo(sender.commands.back());
 
-    _distributor->enableClusterState(lib::ClusterState("storage:0 distributor:1"));
+    enableDistributorClusterState("storage:0 distributor:1");
     CPPUNIT_ASSERT(sender.replies.empty());
     replyToGet(*cb, sender, 2, 110);
 
@@ -992,8 +992,7 @@ TwoPhaseUpdateOperationTest::testUpdateFailsIfOwnershipChangesBetweenGetAndPut()
     // this new state, the distributor no longer owns the bucket in question
     // and the operation should thus be failed. We must not try to send Puts
     // to a bucket we no longer own.
-    _distributor->enableClusterState(
-            lib::ClusterState("storage:2 distributor:1 .0.s:d"));
+    enableDistributorClusterState("storage:2 distributor:1 .0.s:d");
     getBucketDatabase().clear();
     replyToGet(*cb, sender, 0, 70);
     replyToGet(*cb, sender, 1, 70);