1 files changed, 2 insertions, 2 deletions
diff --git a/jstests/replsets/drain.js b/jstests/replsets/drain.js
index a81d45f1d10..74c5868b821 100644
--- a/jstests/replsets/drain.js
+++ b/jstests/replsets/drain.js
@@ -43,7 +43,7 @@
     }
     assert.writeOK(bulk.execute());
     jsTestLog('Number of documents inserted into collection on primary: ' + numDocuments);
-    assert.eq(numDocuments, primary.getDB("foo").foo.count());
+    assert.eq(numDocuments, primary.getDB("foo").foo.find().itcount());
 
     assert.soon(function() {
         var serverStatus = secondary.getDB('foo').serverStatus();
@@ -105,5 +105,5 @@
     // Check for at least two entries. There was one prior to freezing op application on the
     // secondary and we cannot guarantee all writes reached the secondary's op queue prior to
     // shutting down the original primary.
-    assert.gte(primary.getDB("foo").foo.count(), 2);
+    assert.gte(primary.getDB("foo").foo.find().itcount(), 2);
 })();