diff --git a/org.springframework.integration/src/test/java/org/springframework/integration/bus/DirectChannelSubscriptionTests.java b/org.springframework.integration/src/test/java/org/springframework/integration/bus/DirectChannelSubscriptionTests.java
index 8285483f8f..fd59c77774 100644
--- a/org.springframework.integration/src/test/java/org/springframework/integration/bus/DirectChannelSubscriptionTests.java
+++ b/org.springframework.integration/src/test/java/org/springframework/integration/bus/DirectChannelSubscriptionTests.java
@@ -104,7 +104,8 @@ public class DirectChannelSubscriptionTests {
 		};
 		consumer.setOutputChannel(targetChannel);
 		SubscribingConsumerEndpoint endpoint = new SubscribingConsumerEndpoint(consumer, sourceChannel);
-		bus.registerEndpoint(endpoint);
+		context.getBeanFactory().registerSingleton("testEndpoint", endpoint);
+		bus.setApplicationContext(context);
 		context.refresh();
 		bus.start();
 		this.sourceChannel.send(new StringMessage("foo"));