1 files changed, 3 insertions, 5 deletions
diff --git a/jstests/sharding/update_shard_key_doc_on_same_shard.js b/jstests/sharding/update_shard_key_doc_on_same_shard.js
index fecba427593..2fc7a0b2402 100644
--- a/jstests/sharding/update_shard_key_doc_on_same_shard.js
+++ b/jstests/sharding/update_shard_key_doc_on_same_shard.js
@@ -7,19 +7,17 @@
 (function() {
 'use strict';
 
+load("jstests/sharding/libs/sharded_transactions_helpers.js");
 load("jstests/sharding/libs/update_shard_key_helpers.js");
 
-const st = new ShardingTest({
-    mongos: 1,
-    shards: {rs0: {nodes: 3}, rs1: {nodes: 3}},
-    shardOptions: {setParameter: {"coordinateCommitReturnImmediatelyAfterPersistingDecision": true}}
-});
+const st = new ShardingTest({mongos: 1, shards: {rs0: {nodes: 3}, rs1: {nodes: 3}}});
 const kDbName = 'db';
 const ns = kDbName + '.foo';
 const mongos = st.s0;
 const shard0 = st.shard0.shardName;
 const shard1 = st.shard1.shardName;
 
+enableCoordinateCommitReturnImmediatelyAfterPersistingDecision(st);
 assert.commandWorked(mongos.adminCommand({enableSharding: kDbName}));
 st.ensurePrimaryShard(kDbName, shard0);