1 files changed, 5 insertions, 3 deletions
diff --git a/storage/src/tests/persistence/common/filestortestfixture.cpp b/storage/src/tests/persistence/common/filestortestfixture.cpp
index b0225992a20..2fd070de394 100644
--- a/storage/src/tests/persistence/common/filestortestfixture.cpp
+++ b/storage/src/tests/persistence/common/filestortestfixture.cpp
@@ -4,10 +4,12 @@
 #include <vespa/storage/persistence/filestorage/filestormanager.h>
 #include <vespa/persistence/dummyimpl/dummypersistence.h>
 #include <tests/persistence/common/filestortestfixture.h>
+#include <tests/common/make_document_bucket.h>
 #include <vespa/persistence/spi/test.h>
 #include <sstream>
 
 using storage::spi::test::makeBucket;
+using storage::test::makeDocumentBucket;
 
 namespace storage {
 
@@ -99,7 +101,7 @@ FileStorTestFixture::TestFileStorComponents::sendDummyGet(
     std::ostringstream id;
     id << "id:foo:testdoctype1:n=" << bid.getId() << ":0";
     std::shared_ptr<api::GetCommand> cmd(
-            new api::GetCommand(bid, document::DocumentId(id.str()), "[all]"));
+            new api::GetCommand(makeDocumentBucket(bid), document::DocumentId(id.str()), "[all]"));
     cmd->setAddress(makeSelfAddress());
     cmd->setPriority(255);
     top.sendDown(cmd);
@@ -113,7 +115,7 @@ FileStorTestFixture::TestFileStorComponents::sendDummyGetDiff(
     nodes.push_back(0);
     nodes.push_back(1);
     std::shared_ptr<api::GetBucketDiffCommand> cmd(
-            new api::GetBucketDiffCommand(bid, nodes, 12345));
+            new api::GetBucketDiffCommand(makeDocumentBucket(bid), nodes, 12345));
     cmd->setAddress(makeSelfAddress());
     cmd->setPriority(255);
     top.sendDown(cmd);
@@ -130,7 +132,7 @@ FileStorTestFixture::TestFileStorComponents::sendPut(
     document::Document::SP doc(
             _fixture._node->getTestDocMan().createDocument("foobar", id.str()));
     std::shared_ptr<api::PutCommand> cmd(
-            new api::PutCommand(bid, doc, timestamp));
+            new api::PutCommand(makeDocumentBucket(bid), doc, timestamp));
     cmd->setAddress(makeSelfAddress());
     top.sendDown(cmd);
 }