diff --git a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java
index 3402d5d..90de5c8 100644
--- a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java
+++ b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java
@@ -51,11 +51,16 @@ public abstract class SpringBootApplicationIntegrationTestsSupport
 	@Override
 	protected ConfigurableApplicationContext newApplicationContext(Class<?>... annotatedClasses) {
 
-		return setApplicationContext(processBeforeRun(processBeforeBuild(new SpringApplicationBuilder(annotatedClasses)
-			.initializers(this::processBeforeRefresh)
-			.web(getWebApplicationType()))
-			.build())
-			.run(getArguments()));
+		return setApplicationContext(processBeforeRun(processBeforeBuild(contextClass(
+			new SpringApplicationBuilder(annotatedClasses)
+				.initializers(this::processBeforeRefresh)
+				.web(getWebApplicationType())))
+				.build())
+				.run(getArguments()));
+	}
+
+	protected SpringApplicationBuilder contextClass(SpringApplicationBuilder builder) {
+		return builder;
 	}
 
 	protected SpringApplicationBuilder processBeforeBuild(SpringApplicationBuilder springApplicationBuilder) {