diff --git a/spring-webflux/src/test/java/org/springframework/web/reactive/socket/AbstractWebSocketIntegrationTests.java b/spring-webflux/src/test/java/org/springframework/web/reactive/socket/AbstractWebSocketIntegrationTests.java
index 2766c1f926..a97f048e02 100644
--- a/spring-webflux/src/test/java/org/springframework/web/reactive/socket/AbstractWebSocketIntegrationTests.java
+++ b/spring-webflux/src/test/java/org/springframework/web/reactive/socket/AbstractWebSocketIntegrationTests.java
@@ -171,9 +171,7 @@ public abstract class AbstractWebSocketIntegrationTests {
 
 		@Bean
 		public WebSocketService webSocketService() {
-			TomcatRequestUpgradeStrategy strategy = new TomcatRequestUpgradeStrategy();
-			strategy.setMaxSessionIdleTimeout(0L);
-			return new HandshakeWebSocketService(strategy);
+			return new HandshakeWebSocketService(getUpgradeStrategy());
 		}
 
 		protected abstract RequestUpgradeStrategy getUpgradeStrategy();