1 files changed, 1 insertions, 8 deletions
diff --git a/src/mongo/s/query/sharded_agg_test_fixture.h b/src/mongo/s/query/sharded_agg_test_fixture.h
index 621a7e368a9..8fc5a618051 100644
--- a/src/mongo/s/query/sharded_agg_test_fixture.h
+++ b/src/mongo/s/query/sharded_agg_test_fixture.h
@@ -97,14 +97,7 @@ public:
 
         // Mock the expected config server queries.
         expectGetDatabase(nss);
-        expectGetCollection(nss, epoch, UUID::gen(), shardKey);
-        expectFindSendBSONObjVector(kConfigHostAndPort, [&]() {
-            std::vector<BSONObj> response;
-            for (auto&& chunk : chunkDistribution) {
-                response.push_back(chunk.toConfigBSON());
-            }
-            return response;
-        }());
+        expectCollectionAndChunksAggregation(nss, epoch, UUID::gen(), shardKey, chunkDistribution);
 
         const auto cm = future.default_timed_get();
         ASSERT(cm->isSharded());