1 files changed, 12 insertions, 10 deletions

diff --git a/jstests/sharding/balance_repl.js b/jstests/sharding/balance_repl.js
index 39c28b46448..a3c9eefdca8 100644
--- a/jstests/sharding/balance_repl.js
+++ b/jstests/sharding/balance_repl.js
@@ -3,7 +3,7 @@
 //
 
 (function() {
-    "use strict";
+    'use strict';
 
     // The mongod secondaries are set to priority 0 and votes 0 to prevent the primaries
     // from stepping down during migrations on slow evergreen builders.
@@ -33,25 +33,27 @@
     }
     assert.writeOK(bulk.execute());
 
-    s.adminCommand({enablesharding: "test"});
+    assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
     s.ensurePrimaryShard('test', 'test-rs0');
-    s.adminCommand({shardcollection: "test.foo", key: {_id: 1}});
+    assert.commandWorked(s.s0.adminCommand({shardcollection: "test.foo", key: {_id: 1}}));
 
-    for (i = 0; i < 20; i++)
-        s.adminCommand({split: "test.foo", middle: {_id: i * 100}});
+    for (i = 0; i < 20; i++) {
+        assert.commandWorked(s.s0.adminCommand({split: "test.foo", middle: {_id: i * 100}}));
+    }
 
     assert.eq(2100, db.foo.find().itcount());
+
     var coll = db.foo;
     coll.setSlaveOk();
+    assert.eq(2100, coll.find().itcount());
 
     var dbPrimaryShardId = s.getPrimaryShardIdForDatabase("test");
     var other = s.config.shards.findOne({_id: {$ne: dbPrimaryShardId}});
 
     for (i = 0; i < 20; i++) {
-        // Needs to waitForDelete because we'll be performing a slaveOk query,
-        // and secondaries don't have a chunk manager so it doesn't know how to
-        // filter out docs it doesn't own.
-        assert(s.adminCommand({
+        // Needs to waitForDelete because we'll be performing a slaveOk query, and secondaries don't
+        // have a chunk manager so it doesn't know how to filter out docs it doesn't own.
+        assert.commandWorked(s.s0.adminCommand({
             moveChunk: "test.foo",
             find: {_id: i * 100},
             to: other._id,
@@ -59,9 +61,9 @@
             writeConcern: {w: 2},
             _waitForDelete: true
         }));
+
         assert.eq(2100, coll.find().itcount());
     }
 
     s.stop();
-
 }());