diff --git a/test/spec/atlas-data-lake-testing/estimatedDocumentCount.json b/test/spec/atlas-data-lake-testing/estimatedDocumentCount.json
index d039a51f06..87b385208d 100644
--- a/test/spec/atlas-data-lake-testing/estimatedDocumentCount.json
+++ b/test/spec/atlas-data-lake-testing/estimatedDocumentCount.json
@@ -15,8 +15,25 @@
         {
           "command_started_event": {
             "command": {
-              "count": "driverdata"
-            }
+              "aggregate": "driverdata",
+              "pipeline": [
+                {
+                  "$collStats": {
+                    "count": {}
+                  }
+                },
+                {
+                  "$group": {
+                    "_id": 1,
+                    "n": {
+                      "$sum": "$count"
+                    }
+                  }
+                }
+              ]
+            },
+            "command_name": "aggregate",
+            "database_name": "test"
           }
         }
       ]
diff --git a/test/spec/atlas-data-lake-testing/estimatedDocumentCount.yml b/test/spec/atlas-data-lake-testing/estimatedDocumentCount.yml
index b4cecc9522..43eee2d317 100644
--- a/test/spec/atlas-data-lake-testing/estimatedDocumentCount.yml
+++ b/test/spec/atlas-data-lake-testing/estimatedDocumentCount.yml
@@ -13,4 +13,9 @@ tests:
       -
         command_started_event:
           command:
-            count: *collection_name
+            aggregate: *collection_name
+            pipeline:
+            - $collStats: { count: {} }
+            - $group: { _id: 1, n: { $sum: $count }}
+            command_name: aggregate
+            database_name: *database_name