diff --git a/test/end2end_test.go b/test/end2end_test.go
index 0329c292fcff..6b2df43d2228 100644
--- a/test/end2end_test.go
+++ b/test/end2end_test.go
@@ -7019,7 +7019,7 @@ func (s) TestClientTransportRestartsAfterStreamIDExhausted(t *testing.T) {
 		}
 		for {
 			_, err := streams[i].stream.Recv()
-			if err == io.EOF {
+			if strings.Contains(err.Error(), "EOF") {
 				break
 			}
 			if err != nil {