diff --git a/mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/transport/WebFluxSseClientTransportTests.java b/mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/transport/WebFluxSseClientTransportTests.java
index d109a32..7ed325f 100644
--- a/mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/transport/WebFluxSseClientTransportTests.java
+++ b/mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/transport/WebFluxSseClientTransportTests.java
@@ -380,7 +380,7 @@ class WebFluxSseClientTransportTests {
 	void testMessageEndpointValidation() throws InvalidSseMessageEndpointException {
 		var uriCaptor = ArgumentCaptor.forClass(URI.class);
 		verify(sseMessageEndpointValidator).validate(uriCaptor.capture(), matches("/message\\?sessionId=[a-z0-9-]+"));
-		assertThat(uriCaptor.getValue().toString()).matches("http://localhost:\\d+/sse");
+		assertThat(uriCaptor.getValue().toString()).matches(host + "/sse");
 	}
 
 	@Test