diff --git a/spring-integration-redis/src/test/java/org/springframework/integration/redis/leader/RedisLockRegistryLeaderInitiatorTests.java b/spring-integration-redis/src/test/java/org/springframework/integration/redis/leader/RedisLockRegistryLeaderInitiatorTests.java
index b99af5398d..1c935dc161 100644
--- a/spring-integration-redis/src/test/java/org/springframework/integration/redis/leader/RedisLockRegistryLeaderInitiatorTests.java
+++ b/spring-integration-redis/src/test/java/org/springframework/integration/redis/leader/RedisLockRegistryLeaderInitiatorTests.java
@@ -129,12 +129,10 @@ public class RedisLockRegistryLeaderInitiatorTests extends RedisAvailableTests {
 		initiator1.setLeaderEventPublisher(new CountingPublisher(new CountDownLatch(1), revoked11,
 				new CountDownLatch(1)));
 
-		initiator1.getContext().yield();
+		initiator1.stop();
 
 		assertThat(revoked11.await(60, TimeUnit.SECONDS)).isTrue();
 		assertThat(initiator1.getContext().isLeader()).isFalse();
-
-		initiator1.stop();
 	}
 
 	private static class CountingPublisher implements LeaderEventPublisher {