The setUpTestMode helper in retryable_internal_transaction_test.js is already waiting for the primary. However, that appears to be not sufficient when the SDAM ReplicaSetMonitor is used. I think we could solve this by explicitly wait for the RSM on the mongoses to detect the new primary, for example, by calling awaitRSClientHosts like in this test.