diff --git a/spring-di-4/pom.xml b/spring-di-4/pom.xml
index 1eec8efcf0..a486b19e51 100644
--- a/spring-di-4/pom.xml
+++ b/spring-di-4/pom.xml
@@ -24,5 +24,15 @@
             <artifactId>spring-boot-starter-test</artifactId>
         </dependency>
     </dependencies>
-
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.springframework.boot</groupId>
+                <artifactId>spring-boot-maven-plugin</artifactId>
+                <configuration>
+                    <mainClass>com.baeldung.registrypostprocessor.RegistryPostProcessorApplication</mainClass>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
 </project>
\ No newline at end of file