diff --git a/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/main/java/org/springframework/boot/gradle/plugin/SpringBootAotPlugin.java b/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/main/java/org/springframework/boot/gradle/plugin/SpringBootAotPlugin.java
index b67b500e27e..0ee159bf1cb 100644
--- a/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/main/java/org/springframework/boot/gradle/plugin/SpringBootAotPlugin.java
+++ b/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/main/java/org/springframework/boot/gradle/plugin/SpringBootAotPlugin.java
@@ -99,10 +99,10 @@ public class SpringBootAotPlugin implements Plugin<Project> {
 					task.getGroupId().set(project.provider(() -> String.valueOf(project.getGroup())));
 					task.getArtifactId().set(project.provider(() -> project.getName()));
 				});
-		project.getTasks().getByName(aotSourceSet.getCompileJavaTaskName(),
-				(compile) -> compile.dependsOn(generateAotSources));
-		project.getTasks().getByName(aotSourceSet.getProcessResourcesTaskName(),
-				(processResources) -> processResources.dependsOn(generateAotSources));
+		project.getTasks().named(aotSourceSet.getCompileJavaTaskName())
+				.configure((compileJava) -> compileJava.dependsOn(generateAotSources));
+		project.getTasks().named(aotSourceSet.getProcessResourcesTaskName())
+				.configure((processResources) -> processResources.dependsOn(generateAotSources));
 	}
 
 }