4 files changed, 5 insertions, 3 deletions
diff --git a/documentapi/src/tests/messages/messages50test.cpp b/documentapi/src/tests/messages/messages50test.cpp
index 161f976c3d5..0346cadbf9b 100644
--- a/documentapi/src/tests/messages/messages50test.cpp
+++ b/documentapi/src/tests/messages/messages50test.cpp
@@ -4,6 +4,7 @@
 #include <vespa/document/bucket/bucketidfactory.h>
 #include <vespa/document/datatype/documenttype.h>
 #include <vespa/document/select/parser.h>
+#include <vespa/document/repo/documenttyperepo.h>
 #include <vespa/document/update/fieldpathupdates.h>
 #include <vespa/documentapi/documentapi.h>
 
diff --git a/documentapi/src/tests/messages/messages52test.cpp b/documentapi/src/tests/messages/messages52test.cpp
index 642fbc2fc41..eaf8bbec8c4 100644
--- a/documentapi/src/tests/messages/messages52test.cpp
+++ b/documentapi/src/tests/messages/messages52test.cpp
@@ -4,6 +4,7 @@
 
 #include "messages52test.h"
 #include <vespa/documentapi/documentapi.h>
+#include <vespa/document/repo/documenttyperepo.h>
 #include <vespa/document/update/fieldpathupdates.h>
 #include <vespa/document/datatype/documenttype.h>
 
diff --git a/documentapi/src/tests/messages/testbase.cpp b/documentapi/src/tests/messages/testbase.cpp
index 8f2729964a6..02c180853d4 100644
--- a/documentapi/src/tests/messages/testbase.cpp
+++ b/documentapi/src/tests/messages/testbase.cpp
@@ -1,6 +1,7 @@
 // Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 
 #include "testbase.h"
+#include <vespa/document/repo/documenttyperepo.h>
 #include <vespa/document/base/testdocrepo.h>
 #include <vespa/vespalib/util/exception.h>
 #include <fcntl.h>
diff --git a/documentapi/src/tests/messages/testbase.h b/documentapi/src/tests/messages/testbase.h
index 2a1bb61c761..e47858cda4b 100644
--- a/documentapi/src/tests/messages/testbase.h
+++ b/documentapi/src/tests/messages/testbase.h
@@ -1,7 +1,6 @@
 // Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 #pragma once
 
-#include <vespa/document/repo/documenttyperepo.h>
 #include <vespa/documentapi/messagebus/documentprotocol.h>
 #include <vespa/documentapi/loadtypes/loadtypeset.h>
 #include <vespa/messagebus/routable.h>
@@ -23,7 +22,7 @@ typedef bool (TestBase::*TEST_METHOD_PT)();
  * tests to it.
  */
 class TestBase : public vespalib::TestApp {
-    const document::DocumentTypeRepo::SP _repo;
+    std::shared_ptr<const document::DocumentTypeRepo> _repo;
 protected:
     const string                  _dataPath;
     LoadTypeSet                        _loadTypes;
@@ -46,7 +45,7 @@ protected:
 
 public:
     const document::DocumentTypeRepo &getTypeRepo() { return *_repo; }
-    const document::DocumentTypeRepo::SP &getTypeRepoSp() { return _repo; }
+    std::shared_ptr<const document::DocumentTypeRepo> &getTypeRepoSp() { return _repo; }
 
     bool testCoverage(const std::vector<uint32_t> &expected, const std::vector<uint32_t> &actual, bool report = false) const;
     bool writeFile(const string &filename, const mbus::Blob& blob) const;