diff --git a/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/test/java/org/springframework/boot/gradle/junit/GradleCompatibilityExtension.java b/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/test/java/org/springframework/boot/gradle/junit/GradleCompatibilityExtension.java
index fc594eb36a..077bcd1b07 100644
--- a/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/test/java/org/springframework/boot/gradle/junit/GradleCompatibilityExtension.java
+++ b/spring-boot-project/spring-boot-tools/spring-boot-gradle-plugin/src/test/java/org/springframework/boot/gradle/junit/GradleCompatibilityExtension.java
@@ -20,6 +20,7 @@ import java.util.Arrays;
 import java.util.List;
 import java.util.stream.Stream;
 
+import org.junit.jupiter.api.TestTemplate;
 import org.junit.jupiter.api.extension.Extension;
 import org.junit.jupiter.api.extension.ExtensionContext;
 import org.junit.jupiter.api.extension.TestTemplateInvocationContext;
@@ -29,9 +30,9 @@ import org.springframework.boot.gradle.testkit.GradleBuild;
 import org.springframework.boot.gradle.testkit.GradleBuildExtension;
 
 /**
- * {@link Extension} that runs {@link org.junit.jupiter.api.TestTemplate templated tests}
- * against multiple versions of Gradle. Test classes using the extension must have a
- * non-private and non-final {@link GradleBuild} field named {@code gradleBuild}.
+ * {@link Extension} that runs {@link TestTemplate templated tests} against multiple
+ * versions of Gradle. Test classes using the extension must have a non-private and
+ * non-final {@link GradleBuild} field named {@code gradleBuild}.
  *
  * @author Andy Wilkinson
  */