1 files changed, 5 insertions, 7 deletions
diff --git a/src/mongo/db/query/query_test_service_context.h b/src/mongo/db/query/query_test_service_context.h
index c52944cf2f5..05a21afef3f 100644
--- a/src/mongo/db/query/query_test_service_context.h
+++ b/src/mongo/db/query/query_test_service_context.h
@@ -30,7 +30,7 @@
 
 #include "mongo/db/client.h"
 #include "mongo/db/logical_session_id.h"
-#include "mongo/db/service_context_noop.h"
+#include "mongo/db/service_context.h"
 
 namespace mongo {
 
@@ -42,20 +42,18 @@ namespace mongo {
 class QueryTestServiceContext {
 public:
     QueryTestServiceContext();
+    ~QueryTestServiceContext();
 
     ServiceContext::UniqueOperationContext makeOperationContext();
 
     ServiceContext::UniqueOperationContext makeOperationContext(LogicalSessionId lsid);
 
     Client* getClient() const;
-
-    ServiceContext* getServiceContext() {
-        return &_serviceContext;
-    }
+    ServiceContext* getServiceContext();
 
 private:
-    ServiceContextNoop _serviceContext;
-    ServiceContext::UniqueClient _uniqueClient;
+    ServiceContext::UniqueServiceContext _service;
+    ServiceContext::UniqueClient _client;
 };
 
 }  // namespace mongo