diff --git a/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/TcpNioConnectionReadTests.java b/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/TcpNioConnectionReadTests.java
index ff7b489c7a..b319eea3bd 100644
--- a/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/TcpNioConnectionReadTests.java
+++ b/spring-integration-ip/src/test/java/org/springframework/integration/ip/tcp/connection/TcpNioConnectionReadTests.java
@@ -511,7 +511,8 @@ public class TcpNioConnectionReadTests {
 
 		assertThat(errorMessageRef.get().getMessage(),
 				anyOf(containsString("Connection is closed"),
-						containsString("Socket closed during message assembly")));
+						containsString("Socket closed during message assembly"),
+						containsString("Stream closed after 2 of 3")));
 
 		assertTrue(semaphore.tryAcquire(10000, TimeUnit.MILLISECONDS));
 		assertTrue(removed.size() > 0);