1 files changed, 3 insertions, 2 deletions
diff --git a/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp b/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp
index 9e13cf2ff7d..4a195514db1 100644
--- a/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp
+++ b/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp
@@ -4,6 +4,7 @@
 #include <vespa/document/datatype/documenttype.h>
 #include <vespa/document/fieldset/fieldsets.h>
 #include <vespa/document/repo/documenttyperepo.h>
+#include <vespa/document/test/make_bucket_space.h>
 #include <vespa/persistence/spi/documentselection.h>
 #include <vespa/persistence/spi/test.h>
 #include <vespa/persistence/spi/test.h>
@@ -21,6 +22,7 @@ using document::BucketSpace;
 using document::Document;
 using document::DocumentId;
 using document::DocumentType;
+using document::test::makeBucketSpace;
 using search::DocumentMetaData;
 using storage::spi::Bucket;
 using storage::spi::BucketChecksum;
@@ -42,7 +44,6 @@ using storage::spi::Selection;
 using storage::spi::Timestamp;
 using storage::spi::UpdateResult;
 using storage::spi::test::makeSpiBucket;
-using storage::spi::test::makeBucketSpace;
 using namespace proton;
 using namespace vespalib;
 
@@ -610,7 +611,7 @@ TEST_F("require that setClusterState() is routed to handlers", SimpleFixture)
 {
     ClusterState state(createClusterState());
 
-    f.engine.setClusterState(state);
+    f.engine.setClusterState(makeBucketSpace(), state);
     EXPECT_EQUAL(&state, f.hset.handler1.lastCalc);
     EXPECT_EQUAL(&state, f.hset.handler2.lastCalc);
 }