2 files changed, 5 insertions, 3 deletions
diff --git a/documentapi/src/tests/policies/testframe.cpp b/documentapi/src/tests/policies/testframe.cpp
index db8a1b48924..8d5b6585529 100644
--- a/documentapi/src/tests/policies/testframe.cpp
+++ b/documentapi/src/tests/policies/testframe.cpp
@@ -1,4 +1,7 @@
 // Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#include <vespa/fastos/fastos.h>
+#include <vespa/log/log.h>
+LOG_SETUP(".testframe");
 
 #include "testframe.h"
 #include <vespa/messagebus/emptyreply.h>
@@ -8,9 +11,6 @@
 #include <vespa/messagebus/testlib/simpleprotocol.h>
 #include <vespa/messagebus/testlib/simplereply.h>
 
-#include <vespa/log/log.h>
-LOG_SETUP(".testframe");
-
 using document::DocumentTypeRepo;
 using namespace documentapi;
 
diff --git a/documentapi/src/tests/policies/testframe.h b/documentapi/src/tests/policies/testframe.h
index f0e37d979f3..e0780733331 100644
--- a/documentapi/src/tests/policies/testframe.h
+++ b/documentapi/src/tests/policies/testframe.h
@@ -197,6 +197,8 @@ public:
      * @return Handle to the system state.
      */
     documentapi::SystemStateHandle getSystemState();
+
+    // Implements IReplyHandler.
     void handleReply(mbus::Reply::UP reply) override;
 };