1 files changed, 7 insertions, 2 deletions
diff --git a/jstests/sharding/sharding_balance4.js b/jstests/sharding/sharding_balance4.js
index cfe5c3b6d67..99c005bd09e 100644
--- a/jstests/sharding/sharding_balance4.js
+++ b/jstests/sharding/sharding_balance4.js
@@ -6,6 +6,7 @@
  * @tags: [resource_intensive]
  */
 (function() {
+load('jstests/sharding/autosplit_include.js');
 
 var s = new ShardingTest({shards: 2, mongos: 1, other: {chunkSize: 1, enableAutoSplit: true}});
 
@@ -64,15 +65,19 @@ Random.setRandomSeed();
 
 // Initially update all documents from 1 to N, otherwise later checks can fail because no
 // document previously existed
-var bulk = db.foo.initializeUnorderedBulkOp();
 for (i = 0; i < N; i++) {
+    let bulk = db.foo.initializeUnorderedBulkOp();
     doUpdate(bulk, true, i);
+    assert.commandWorked(bulk.execute());
+    waitForOngoingChunkSplits(s);
 }
 
 for (i = 0; i < N * 9; i++) {
+    let bulk = db.foo.initializeUnorderedBulkOp();
     doUpdate(bulk, false);
+    assert.commandWorked(bulk.execute());
+    waitForOngoingChunkSplits(s);
 }
-assert.commandWorked(bulk.execute());
 
 for (var i = 0; i < 50; i++) {
     s.printChunks("test.foo");