diff --git a/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java b/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java
index fe97f4e81f..d90af2e0c8 100644
--- a/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java
+++ b/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java
@@ -94,13 +94,13 @@ class ConfigurationClassPostProcessorAotContributionTests {
 	@Test
 	void applyToWhenHasImportAwareConfigurationRegistersBeanPostProcessorAfterApplicationContextAwareProcessor() {
 		BeanFactoryInitializationAotContribution contribution = getContribution(
-				ImportConfiguration.class);
+				TestAwareCallbackConfiguration.class);
 		contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode);
 		compile((initializer, compiled) -> {
-			GenericApplicationContext freshContext = new AnnotationConfigApplicationContext();
+			GenericApplicationContext freshContext = new GenericApplicationContext();
 			DefaultListableBeanFactory freshBeanFactory = freshContext.getDefaultListableBeanFactory();
 			initializer.accept(freshBeanFactory);
-			freshContext.registerBean(TestAwareCallbackConfiguration.class);
+			freshContext.registerBean(TestAwareCallbackBean.class);
 			freshContext.refresh();
 			TestAwareCallbackBean bean = freshContext.getBean(TestAwareCallbackBean.class);
 			assertThat(bean.instances).hasSize(2);