diff --git a/build.gradle b/build.gradle
index e082639e..ef232545 100644
--- a/build.gradle
+++ b/build.gradle
@@ -916,11 +916,11 @@ project('tcp-broadcast') {
 	}
 
 	bootRun {
-		main = 'org.springframework.integration.samples.tcpbroadcast.Application'
+		main = 'org.springframework.integration.samples.tcpbroadcast.TcpBroadcastApplication'
 	}
 
 	task run(type: JavaExec) {
-		main 'org.springframework.integration.samples.tcpbroadcast.Application'
+		main 'org.springframework.integration.samples.tcpbroadcast.TcpBroadcastApplication'
 		classpath = sourceSets.main.runtimeClasspath
 	}
 }