Simplify distributor unit tests.

1 files changed, 2 insertions, 3 deletions

diff --git a/storage/src/tests/distributor/removeoperationtest.cpp b/storage/src/tests/distributor/removeoperationtest.cpp
index 423e0816c13..24c40f50894 100644
--- a/storage/src/tests/distributor/removeoperationtest.cpp
+++ b/storage/src/tests/distributor/removeoperationtest.cpp
@@ -46,7 +46,7 @@ public:
 
         docId = document::DocumentId(document::DocIdString("test", "uri"));
         bucketId = getExternalOperationHandler().getBucketId(docId);
-        _distributor->enableClusterState(lib::ClusterState("distributor:1 storage:4"));
+        enableDistributorClusterState("distributor:1 storage:4");
     };
 
     void tearDown() override {
@@ -189,8 +189,7 @@ RemoveOperationTest::testMultipleCopies()
 void
 RemoveOperationTest::canSendRemoveWhenAllReplicaNodesRetired()
 {
-    _distributor->enableClusterState(
-            lib::ClusterState("distributor:1 storage:1 .0.s:r"));
+    enableDistributorClusterState("distributor:1 storage:1 .0.s:r");
     addNodesToBucketDB(bucketId, "0=123");
     sendRemove();