diff --git a/spring-boot-tools/spring-boot-gradle-plugin/src/main/groovy/org/springframework/boot/gradle/agent/AgentTasksEnhancer.java b/spring-boot-tools/spring-boot-gradle-plugin/src/main/groovy/org/springframework/boot/gradle/agent/AgentTasksEnhancer.java
index 89f356d1f5..0ffdca2cf1 100644
--- a/spring-boot-tools/spring-boot-gradle-plugin/src/main/groovy/org/springframework/boot/gradle/agent/AgentTasksEnhancer.java
+++ b/spring-boot-tools/spring-boot-gradle-plugin/src/main/groovy/org/springframework/boot/gradle/agent/AgentTasksEnhancer.java
@@ -85,14 +85,12 @@ public class AgentTasksEnhancer implements Action<Project> {
 			if (loaded != null) {
 				CodeSource source = loaded.getProtectionDomain().getCodeSource();
 				if (source != null) {
-					File agent;
 					try {
-						agent = new File(source.getLocation().toURI());
+						return new File(source.getLocation().toURI());
 					}
 					catch (URISyntaxException ex) {
-						agent = new File(source.getLocation().getPath());
+						return new File(source.getLocation().getPath());
 					}
-					return agent;
 				}
 			}
 		}