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 f384f9ca86..e8213e7b5b 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
@@ -531,7 +531,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);