diff --git a/buildSrc/src/main/java/org/springframework/boot/build/architecture/ArchitecturePlugin.java b/buildSrc/src/main/java/org/springframework/boot/build/architecture/ArchitecturePlugin.java
index 4c73e021df..44bf79a32d 100644
--- a/buildSrc/src/main/java/org/springframework/boot/build/architecture/ArchitecturePlugin.java
+++ b/buildSrc/src/main/java/org/springframework/boot/build/architecture/ArchitecturePlugin.java
@@ -22,7 +22,7 @@ import java.util.List;
 import org.gradle.api.Plugin;
 import org.gradle.api.Project;
 import org.gradle.api.Task;
-import org.gradle.api.plugins.JavaBasePlugin;
+import org.gradle.api.plugins.JavaPlugin;
 import org.gradle.api.plugins.JavaPluginExtension;
 import org.gradle.api.tasks.SourceSet;
 import org.gradle.api.tasks.TaskProvider;
@@ -39,7 +39,7 @@ public class ArchitecturePlugin implements Plugin<Project> {
 
 	@Override
 	public void apply(Project project) {
-		project.getPlugins().withType(JavaBasePlugin.class, (javaPlugin) -> registerTasks(project));
+		project.getPlugins().withType(JavaPlugin.class, (javaPlugin) -> registerTasks(project));
 	}
 
 	private void registerTasks(Project project) {