diff --git a/spring-context/src/main/kotlin/org/springframework/context/support/GenericApplicationContextExtension.kt b/spring-context/src/main/kotlin/org/springframework/context/support/GenericApplicationContextExtension.kt
index 55474ed4b1..1bf3a29680 100644
--- a/spring-context/src/main/kotlin/org/springframework/context/support/GenericApplicationContextExtension.kt
+++ b/spring-context/src/main/kotlin/org/springframework/context/support/GenericApplicationContextExtension.kt
@@ -1,6 +1,7 @@
 package org.springframework.context.support
 
 import org.springframework.beans.factory.config.BeanDefinitionCustomizer
+import org.springframework.context.ApplicationContext
 import java.util.function.Supplier
 import kotlin.reflect.KClass
 
@@ -34,7 +35,7 @@ object GenericApplicationContextExtension {
 	 * @see GenericApplicationContext.registerBean(Class<T>, Supplier<T>, BeanDefinitionCustomizer...)
 	 */
 	inline fun <reified T : Any> GenericApplicationContext.registerBean(
-			vararg customizers: BeanDefinitionCustomizer, crossinline function: (GenericApplicationContext) -> T) {
+			vararg customizers: BeanDefinitionCustomizer, crossinline function: (ApplicationContext) -> T) {
 					registerBean(T::class.java, Supplier { function.invoke(this) }, *customizers)
 	}
 
@@ -42,7 +43,7 @@ object GenericApplicationContextExtension {
 	 * @see GenericApplicationContext.registerBean(String, Class<T>, Supplier<T>, BeanDefinitionCustomizer...)
 	 */
 	inline fun <reified T : Any> GenericApplicationContext.registerBean(name: String,
-			vararg customizers: BeanDefinitionCustomizer, crossinline function: (GenericApplicationContext) -> T) {
+			vararg customizers: BeanDefinitionCustomizer, crossinline function: (ApplicationContext) -> T) {
 					registerBean(name, T::class.java, Supplier { function.invoke(this) }, *customizers)
 	}
 }