SERVER-39278 Wait for rollback to finish before ending session in transient_txn_error_labels_with_write_concern.js

1 files changed, 1 insertions, 1 deletions

diff --git a/jstests/replsets/transient_txn_error_labels_with_write_concern.js b/jstests/replsets/transient_txn_error_labels_with_write_concern.js
index 5c16667e7c4..6b11967cfb2 100644
--- a/jstests/replsets/transient_txn_error_labels_with_write_concern.js
+++ b/jstests/replsets/transient_txn_error_labels_with_write_concern.js
@@ -102,6 +102,7 @@
 
     jsTest.log("NoSuchTransaction without write concern error is transient");
     restartServerReplication(rst.nodes);
+    rst.awaitReplication();
     res = newPrimarySessionDb.adminCommand({
         commitTransaction: 1,
         txnNumber: NumberLong(txnNumber),
@@ -112,7 +113,6 @@
     assert(!res.hasOwnProperty("writeConcernError"), res);
     assert.eq(res["errorLabels"], ["TransientTransactionError"], res);
 
-    rst.awaitNodesAgreeOnPrimary();
     session.endSession();
 
     rst.stopSet();