diff --git a/spring-core/src/main/java/org/springframework/aot/generate/DefaultGenerationContext.java b/spring-core/src/main/java/org/springframework/aot/generate/DefaultGenerationContext.java
index 23de9c5b2a..e087d0fef0 100644
--- a/spring-core/src/main/java/org/springframework/aot/generate/DefaultGenerationContext.java
+++ b/spring-core/src/main/java/org/springframework/aot/generate/DefaultGenerationContext.java
@@ -115,7 +115,7 @@ public class DefaultGenerationContext implements GenerationContext {
 	}
 
 	@Override
-	public GenerationContext withName(String name) {
+	public DefaultGenerationContext withName(String name) {
 		return new DefaultGenerationContext(this, name);
 	}